Update app.py

app.py

@@ -39,8 +39,24 @@ The model was trained on Thai audio recordings with the following sentences, so
 <img src="https://huggingface.co/spaces/course-demos/Rick_and_Morty_QA/resolve/main/rick.png" width=200px>
 """
 
+def actualpredict(file_path):
+    model.eval()
+    with torch.no_grad():
+        wav_data, _ = sf.read(file_path.name)
+        inputs = processor(wav_data, sampling_rate=16000, return_tensors="pt", padding=True)
 
+        input_values = inputs.input_values.squeeze(0)  
+        if max_length - input_values.shape[-1] > 0:
+            input_values = torch.cat([input_values, torch.zeros((max_length - input_values.shape[-1],))], dim=-1)
+        else:
+            input_values = input_values[:max_length]
+        input_values = input_values.unsqueeze(0).to(device)
+        inputs = {"input_values": input_values}
 
+        logits = model(**inputs).logits
+        logits = logits.squeeze()
+        predicted_class_id = torch.argmax(logits, dim=-1).item()
+    return predicted_class_id
 
 
 def predict(microphone,file_upload):
@@ -62,22 +78,7 @@ def predict(microphone,file_upload):
     if(file_upload is not None):
         file_path = file_upload
 
-    model.eval()
-    with torch.no_grad():
-        wav_data, _ = sf.read(file_path.name)
-        inputs = processor(wav_data, sampling_rate=16000, return_tensors="pt", padding=True)
-
-        input_values = inputs.input_values.squeeze(0)  
-        if max_length - input_values.shape[-1] > 0:
-            input_values = torch.cat([input_values, torch.zeros((max_length - input_values.shape[-1],))], dim=-1)
-        else:
-            input_values = input_values[:max_length]
-        input_values = input_values.unsqueeze(0).to(device)
-        inputs = {"input_values": input_values}
-
-        logits = model(**inputs).logits
-        logits = logits.squeeze()
-        predicted_class_id = torch.argmax(logits, dim=-1).item()
+    predicted_class_id = actualpredict(file_path)
     if(predicted_class_id==0):
         ans = "no_parkinson"
     else: