Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ def predict(inp):
|
|
| 30 |
prediction = torch.nn.functional.softmax(model(inp)[0], dim=0)
|
| 31 |
return {LABELS[i]: float(prediction[i]) for i in range(2)}
|
| 32 |
|
| 33 |
-
examples = [['VWUP.jpg']]
|
| 34 |
|
| 35 |
iface = gr.Interface(predict, inputs=gr.inputs.Image(), outputs="label", title=title, description=description, examples=examples)
|
| 36 |
iface.launch()
|
|
|
|
| 30 |
prediction = torch.nn.functional.softmax(model(inp)[0], dim=0)
|
| 31 |
return {LABELS[i]: float(prediction[i]) for i in range(2)}
|
| 32 |
|
| 33 |
+
examples = [['VWUP.jpg'],['fiat500.jpg']]
|
| 34 |
|
| 35 |
iface = gr.Interface(predict, inputs=gr.inputs.Image(), outputs="label", title=title, description=description, examples=examples)
|
| 36 |
iface.launch()
|