Spaces:
Runtime error
Runtime error
Steven Anderson
commited on
Commit
•
8378f45
1
Parent(s):
c9db301
Fix
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ def process(image, prompt):
|
|
| 24 |
probs = logits_per_image.softmax(dim=-1).cpu().numpy()
|
| 25 |
print(probs)
|
| 26 |
|
| 27 |
-
return dict(zip(prompts, probs
|
| 28 |
|
| 29 |
|
| 30 |
iface = gr.Interface(
|
|
|
|
| 24 |
probs = logits_per_image.softmax(dim=-1).cpu().numpy()
|
| 25 |
print(probs)
|
| 26 |
|
| 27 |
+
return dict(zip(prompts, probs))
|
| 28 |
|
| 29 |
|
| 30 |
iface = gr.Interface(
|