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=[
|