Update app.py
Browse files
app.py
CHANGED
@@ -43,8 +43,19 @@ The model was trained on Thai audio recordings with the following sentences:
|
|
43 |
|
44 |
|
45 |
|
46 |
-
def predict(
|
|
|
47 |
max_length = 100000
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
48 |
|
49 |
model.eval()
|
50 |
with torch.no_grad():
|
@@ -66,7 +77,7 @@ def predict(file_path):
|
|
66 |
ans = "no_parkinson"
|
67 |
else:
|
68 |
ans = "parkinson"
|
69 |
-
return ans
|
70 |
gr.Interface(
|
71 |
fn=predict,
|
72 |
inputs=[
|
|
|
43 |
|
44 |
|
45 |
|
46 |
+
def predict(microphone,file_upload):
|
47 |
+
|
48 |
max_length = 100000
|
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. "
|
53 |
+
"The recorded file from the microphone will be used and the uploaded audio will be discarded.\n"
|
54 |
+
)
|
55 |
+
|
56 |
+
elif (microphone is None) and (file_upload is None):
|
57 |
+
return "ERROR: You have to either use the microphone or upload an audio file"
|
58 |
+
file_path = microphone if microphone is not None else file_upload
|
59 |
|
60 |
model.eval()
|
61 |
with torch.no_grad():
|
|
|
77 |
ans = "no_parkinson"
|
78 |
else:
|
79 |
ans = "parkinson"
|
80 |
+
return warn_output + ans
|
81 |
gr.Interface(
|
82 |
fn=predict,
|
83 |
inputs=[
|