spycoder commited on
Commit
5549008
1 Parent(s): 33a5bcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -64,10 +64,13 @@ def predict(file_path):
64
  predicted_class_id = torch.argmax(logits, dim=-1).item()
65
 
66
  return predicted_class_id
67
- gr.Interface(
68
- fn=predict,
69
- inputs="file",
70
- outputs="text",
71
- title=title,
72
- description=description,
73
- ).launch()
 
 
 
 
64
  predicted_class_id = torch.argmax(logits, dim=-1).item()
65
 
66
  return predicted_class_id
67
+ # gr.Interface(
68
+ # fn=predict,
69
+ # inputs="file",
70
+ # outputs="text",
71
+ # title=title,
72
+ # description=description,
73
+ # ).launch()
74
+
75
+ iface = gr.Interface(fn=predict, inputs="file", outputs="text")
76
+ iface.launch()