Update app.py
Browse files
app.py
CHANGED
|
@@ -140,7 +140,7 @@ def predict(user_input,company):
|
|
| 140 |
|
| 141 |
# Set-up the Gradio UI
|
| 142 |
user_input = gr.Textbox (label = 'Query')
|
| 143 |
-
|
| 144 |
['aws','google','IBM','Meta','msft'],
|
| 145 |
label = 'company'
|
| 146 |
)
|
|
@@ -157,7 +157,7 @@ model_output = gr.Textbox (label = 'Response')
|
|
| 157 |
|
| 158 |
demo = gr.Interface(
|
| 159 |
fn=predict,
|
| 160 |
-
inputs=[user_input,
|
| 161 |
outputs=[model_output],
|
| 162 |
title="RAG on 10k-reports",
|
| 163 |
description="This API allows you to query on annaul reports",
|
|
|
|
| 140 |
|
| 141 |
# Set-up the Gradio UI
|
| 142 |
user_input = gr.Textbox (label = 'Query')
|
| 143 |
+
company = gr.Radio(
|
| 144 |
['aws','google','IBM','Meta','msft'],
|
| 145 |
label = 'company'
|
| 146 |
)
|
|
|
|
| 157 |
|
| 158 |
demo = gr.Interface(
|
| 159 |
fn=predict,
|
| 160 |
+
inputs=[user_input,company],
|
| 161 |
outputs=[model_output],
|
| 162 |
title="RAG on 10k-reports",
|
| 163 |
description="This API allows you to query on annaul reports",
|