Upload app.py

app.py

@@ -139,7 +139,7 @@ def predict(user_input,company):
             ))
             f.write("\n")
 
-    return prediction
+    return prediction [0]
 
 # Set-up the Gradio UI
 user_input = gr.Textbox (label = 'Query')
@@ -161,7 +161,7 @@ model_output = gr.Label (label = 'Response')
 demo = gr.Interface(
     fn=predict,
     inputs=[user_input,company_input],
-    outputs=prediction,
+    outputs=model_output,
     title="RAG on 10k-reports",
     description="This API allows you to query on annaul reports",
     concurrency_limit=16