Spaces:

whilefalse
/

CLIP

Runtime error

Steven Anderson commited on Apr 13, 2023

Commit

8378f45

1 Parent(s): c9db301

Fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ def process(image, prompt):
         probs = logits_per_image.softmax(dim=-1).cpu().numpy()
         print(probs)
-        return dict(zip(prompts, probs[0]))
 iface = gr.Interface(

         probs = logits_per_image.softmax(dim=-1).cpu().numpy()
         print(probs)
+        return dict(zip(prompts, probs))
 iface = gr.Interface(