Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from transformers import pipeline
-from transformers.pipelines import PipelineException
 
 # Preload models
 models = {
@@ -15,18 +14,13 @@ loaded_models = {}
 
 def load_model(model_name):
     if model_name not in loaded_models:
-        try:
-            loaded_models[model_name] = pipeline("question-answering", model=models[model_name])
-        except PipelineException as e:
-            return str(e)
+        loaded_models[model_name] = pipeline("question-answering", model=models[model_name])
     return loaded_models[model_name]
 
 def answer_question(model_name, file, question):
     model = load_model(model_name)
-    if isinstance(model, str):
-        return model, "", ""
     
-    context = file.read() if file else ""
+    context = file.read().decode('utf-8') if file else ""
     result = model(question=question, context=context)
     answer = result['answer']
     score = result['score']
@@ -63,15 +57,19 @@ with gr.Blocks() as interface:
     with gr.Row():
         submit_button = gr.Button("Submit")
     
-    gr.Markdown("### Progress")
-    with gr.Row():
-        progress_bar = gr.Progress(label="Loading Model...")
+    # Define a status area for progress
+    status = gr.Markdown(value="")
+
+    def on_submit(model_name, file, question):
+        status.update(value="Loading model...")
+        answer, score, explanation = answer_question(model_name, file, question)
+        status.update(value="Model loaded")
+        return answer, score, explanation
 
     submit_button.click(
-        answer_question,
+        on_submit,
         inputs=[model_dropdown, file_input, question_input],
-        outputs=[answer_output, score_output, explanation_output],
-        show_progress=progress_bar,
+        outputs=[answer_output, score_output, explanation_output]
     )
 
 if __name__ == "__main__":