spycoder commited on
Commit
61137cb
1 Parent(s): 4345351

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ def predict(file_upload,microphone):
77
  logits = logits.squeeze()
78
  predicted_class_id = torch.argmax(logits, dim=-1).item()
79
 
80
- return warn_output + "You probably have SP" if predicted_class_id == 1 else warn_output + "You probably don't have SP"
81
  gr.Interface(
82
  fn=predict,
83
  inputs=[
 
77
  logits = logits.squeeze()
78
  predicted_class_id = torch.argmax(logits, dim=-1).item()
79
 
80
+ return warn_output + "You probably have PP" if predicted_class_id == 1 else warn_output + "You probably don't have PP"
81
  gr.Interface(
82
  fn=predict,
83
  inputs=[