Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,10 +30,10 @@ def predict(image: Image.Image):
|
|
| 30 |
predicted = torch.argmax(logits, dim=1).item()
|
| 31 |
return config.id2label[str(predicted)]
|
| 32 |
|
| 33 |
-
#
|
| 34 |
gr.Interface(
|
| 35 |
fn=predict,
|
| 36 |
inputs=gr.Image(type="pil"),
|
| 37 |
outputs=gr.Label(),
|
| 38 |
-
|
| 39 |
-
).launch(share=True
|
|
|
|
| 30 |
predicted = torch.argmax(logits, dim=1).item()
|
| 31 |
return config.id2label[str(predicted)]
|
| 32 |
|
| 33 |
+
# ✅ Prediction interface
|
| 34 |
gr.Interface(
|
| 35 |
fn=predict,
|
| 36 |
inputs=gr.Image(type="pil"),
|
| 37 |
outputs=gr.Label(),
|
| 38 |
+
flagging_mode="never"
|
| 39 |
+
).queue().launch(share=True)
|