Fix

app.py

@@ -4,7 +4,7 @@ import clip
 from PIL import Image
 
 
-def process(device, image, prompt):
+def process(model, preprocess, device, image, prompt):
     print("Inferring...")
     image = preprocess(image).unsqueeze(0).to(device)
     print(image)
@@ -21,19 +21,18 @@ def process(device, image, prompt):
         return dict(zip(prompts, probs[0]))
 
 
-if __name__ == "__main__":
-    print("Getting device...")
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    print("Loading model...")
-    model, preprocess = clip.load("ViT-B/32", device=device)
-    print("Loaded model.")
-
-    iface = gr.Interface(
-        fn=lambda i, p: process(device, i, p),
-        inputs=[
-            gr.Image(),
-            gr.Textbox(lines=5, label="Prompts (newline-separated)"),
-        ],
-        outputs="label",
-    )
-    iface.launch()
+print("Getting device...")
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print("Loading model...")
+model, preprocess = clip.load("ViT-B/32", device=device)
+print("Loaded model.")
+
+iface = gr.Interface(
+    fn=lambda i, p: process(model, preprocess, device, i, p),
+    inputs=[
+        gr.Image(),
+        gr.Textbox(lines=5, label="Prompts (newline-separated)"),
+    ],
+    outputs="label",
+)
+iface.launch()