ariankhalfani commited on
Commit
173d79d
1 Parent(s): 9874be5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -108,11 +108,11 @@ def upload_pdf(file):
108
  # Gradio interface
109
  iface = gr.Interface(
110
  fn=chatbot,
111
- inputs=[gr.inputs.Dropdown(["Meta-Llama-3-70B-Instruct", "Meta-Llama-3-8B-Instruct", "Gemma-2-27B-IT", "Gemma-2-27B"]), "text"],
112
- outputs="text",
113
  title="Storage Warehouse Customer Service Chatbot"
114
  )
115
- file_upload = gr.Interface(fn=upload_pdf, inputs="file", outputs="text", title="Upload PDF for Context")
116
 
117
  app = gr.TabbedInterface([iface, file_upload], ["Chatbot", "Upload PDF"])
118
  app.launch(share=True)
 
108
  # Gradio interface
109
  iface = gr.Interface(
110
  fn=chatbot,
111
+ inputs=[gr.Dropdown(["Meta-Llama-3-70B-Instruct", "Meta-Llama-3-8B-Instruct", "Gemma-2-27B-IT", "Gemma-2-27B"]), gr.Textbox()],
112
+ outputs=gr.Textbox(),
113
  title="Storage Warehouse Customer Service Chatbot"
114
  )
115
+ file_upload = gr.Interface(fn=upload_pdf, inputs=gr.File(), outputs=gr.Textbox(), title="Upload PDF for Context")
116
 
117
  app = gr.TabbedInterface([iface, file_upload], ["Chatbot", "Upload PDF"])
118
  app.launch(share=True)