Update app.py

app.py

@@ -15,33 +15,15 @@ def model_performance_UQS(model_checkpoint, context, question):
 
 m1 = gr.Interface.load("huggingface/amitjohn007/albert-finetuned-squad")
 m2 = gr.Interface.load("huggingface/amitjohn007/bert-finetuned-squad")
-m3 = gr.Interface.load("huggingface/amitjohn007/roberta-base-finetuned-squad")
-m4 = gr.Interface.load("huggingface/amitjohn007/xlm-roberta-base-finetuned-squad")
-m5 = gr.Interface.load("huggingface/amitjohn007/simplebert-base-finetuned-squad")
-m6 = gr.Interface.load("huggingface/amitjohn007/electra-finetuned-squad")
-m7 = gr.Interface.load("huggingface/amitjohn007/second-mobil-bert-finetuned-squad")
-m8 = gr.Interface.load("huggingface/amitjohn007/mpnet-finetuned")
 
 def predict(model_choice,context,question):
   if model_choice=="m1":   
-    return model_performance_UQS(model_checkpoint="huggingface/amitjohn007/albert-finetuned-squad",context=context,question=question) 
-  elif model_choice==m2:
-    return model_performance_UQS(model_checkpoint=m2,context=context,question=question)
-  elif model_choice=="m3":
-    return model_performance_UQS(model_checkpoint=m3,context=context,question=question) 
-  elif model_choice=="m4":
-    return model_performance_UQS(model_checkpoint=m4,context=context,question=question) 
-  elif model_choice=="m5":
-    return model_performance_UQS(model_checkpoint=m5,context=context,question=question) 
-  elif model_choice=="m6":
-    return model_performance_UQS(model_checkpoint=m6,context=context,question=question) 
-  elif model_choice=="m7":
-    return model_performance_UQS(model_checkpoint=m7,context=context,question=question) 
-  elif model_choice=="m8":
-    return model_performance_UQS(model_checkpoint=m8,context=context,question=question) 
+    return model_performance_UQS(model_checkpoint=m1,context=context,question=question) 
+  elif model_choice=="m2":
+    return model_performance_UQS(model_checkpoint=m2,context=context,question=question) 
 
 Interface=gr.Interface(
         fn = predict,
-        inputs = [gr.inputs.Dropdown(["m1","m2","m3","m4","m5","m6","m7","m8"],"value"),gr.inputs.Textbox(lines=7, default=context, label="Context Paragraph"), gr.inputs.Textbox(lines=2, default=question, label="Question")],
+        inputs = [gr.inputs.Dropdown(["m1","m2"],"value"),gr.inputs.Textbox(lines=7, default=context, label="Context Paragraph"), gr.inputs.Textbox(lines=2, default=question, label="Question")],
         outputs = [gr.outputs.Textbox(label="Answer"), gr.outputs.Textbox(label="Score")]
 ).launch(debug=True)