Upload app.py

app.py

@@ -137,18 +137,17 @@ def predict(user_input,company):
 
 # Set-up the Gradio UI
 user_input = gr.Textbox (label = 'Query')
-company_input = gr.Dropdown(
+company_input = gr.Radio(
     ['aws','google','IBM','Meta','msft'], 
     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.
 
-textbox = gr.Textbox()
-company = gr.Radio()
+
 
 # Create the interface
 # For the inputs parameter of Interface provide [textbox,company]
@@ -156,7 +155,7 @@ company = gr.Radio()
 demo = gr.Interface(
     fn=predict,
     inputs=[user_input,company],
-    outputs=model_response,
+    outputs=model_output,
     title="RAG on 10k-reports",
     description="This API allows you to query on annaul reports",
     concurrency_limit=16