Update app.py

app.py

@@ -56,7 +56,14 @@ def get_sos_status(transcription, key_phrase):
         sos = 0
     return sos
 
-def main(audio):
+def main(file , micro=None):
+    if file is not None and micro is None:
+        audio = file
+    elif file is None and micro is not None:
+        audio = micro
+    else:
+        print("THERE IS A PROBLEM")
+        input_audio = file
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     SPT_MODEL = "./SPT_model"
     model_name_or_path = "./SER_model"
@@ -93,8 +100,8 @@ def main(audio):
 
 gr.Interface(
     fn=main, 
-    inputs=[gr.Audio(source="upload", type="filepath"),
-        gr.Audio(source="microphone", type="filepath", streaming=False)],
+    inputs=[gr.Audio(source="upload", type="filepath", label = "File"),
+        gr.Audio(source="microphone", type="filepath", streaming=False, label = "Microphone")],
     outputs=[
         "textbox"
     ],