Update app.py
Browse files
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
|
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=[
|