Update app.py

app.py

@@ -136,7 +136,7 @@ def predict(user_input,company):
             ))
             f.write("\n")
 
-    return prediction [0]
+    return prediction
 
 # Set-up the Gradio UI
 user_input = gr.Textbox (label = 'Query')
@@ -145,7 +145,7 @@ company = gr.Radio(
     label = 'company'
     )
 
-model_output = gr.Textbox (label = 'Response')
+
 
 # Add text box and radio button to the interface
 # The radio button is used to select the company 10k report in which the context needs to be retrieved.
@@ -158,7 +158,7 @@ model_output = gr.Textbox (label = 'Response')
 demo = gr.Interface(
     fn=predict,
     inputs=[user_input,company],
-    outputs=[model_output],
+    outputs="text",
     title="RAG on 10k-reports",
     description="This API allows you to query on annaul reports",
     concurrency_limit=16