Refactor

app.py

@@ -3,8 +3,14 @@ import torch
 import clip
 from PIL import Image
 
+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.")
+
 
-def process(model, preprocess, device, image, prompt):
+def process(image, prompt):
     print("Inferring...")
     image = preprocess(image).unsqueeze(0).to(device)
     print(image)
@@ -21,14 +27,8 @@ def process(model, preprocess, device, image, prompt):
         return dict(zip(prompts, probs[0]))
 
 
-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),
+    fn=process,
     inputs=[
         gr.Image(),
         gr.Textbox(lines=5, label="Prompts (newline-separated)"),