Update app.py

app.py

@@ -62,13 +62,16 @@ def predict(file_path):
         logits = model(**inputs).logits
         logits = logits.squeeze()
         predicted_class_id = torch.argmax(logits, dim=-1).item()
-
-    return predicted_class_id
+    if(predicted_class_id==0):
+        ans = "no_parkinson"
+    else:
+        ans = "parkinson"
+    return ans
 gr.Interface(
     fn=predict,
     inputs=[
-        gr.inputs.Audio(source="microphone", type="filepath", optional=True),
-        gr.inputs.Audio(source="upload", type="filepath", optional=True),
+        gr.inputs.Audio(source="microphone", type="file", optional=True),
+        gr.inputs.Audio(source="upload", type="file", optional=True),
     ],
     outputs="text",
     title=title,