Update app.py
Browse files
app.py
CHANGED
|
@@ -46,6 +46,7 @@ The model was trained on Thai audio recordings with the following sentences: \n
|
|
| 46 |
def predict(file_upload,microphone):
|
| 47 |
max_length = 100000
|
| 48 |
file_path =file_upload
|
|
|
|
| 49 |
if (microphone is not None) and (file_upload is not None):
|
| 50 |
warn_output = (
|
| 51 |
"WARNING: You've uploaded an audio file and used the microphone. "
|
|
@@ -75,7 +76,7 @@ def predict(file_upload,microphone):
|
|
| 75 |
logits = logits.squeeze()
|
| 76 |
predicted_class_id = torch.argmax(logits, dim=-1).item()
|
| 77 |
|
| 78 |
-
return "You probably have SP" if predicted_class_id == 1 else "You probably don't have SP"
|
| 79 |
gr.Interface(
|
| 80 |
fn=predict,
|
| 81 |
inputs=[
|
|
|
|
| 46 |
def predict(file_upload,microphone):
|
| 47 |
max_length = 100000
|
| 48 |
file_path =file_upload
|
| 49 |
+
warn_output = " "
|
| 50 |
if (microphone is not None) and (file_upload is not None):
|
| 51 |
warn_output = (
|
| 52 |
"WARNING: You've uploaded an audio file and used the microphone. "
|
|
|
|
| 76 |
logits = logits.squeeze()
|
| 77 |
predicted_class_id = torch.argmax(logits, dim=-1).item()
|
| 78 |
|
| 79 |
+
return warn_output + "\n" + "You probably have SP" if predicted_class_id == 1 else "You probably don't have SP"
|
| 80 |
gr.Interface(
|
| 81 |
fn=predict,
|
| 82 |
inputs=[
|