syedMohib44 commited on
Commit
e90608d
·
1 Parent(s): c5091e3
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -8,6 +8,9 @@ from fastapi import FastAPI, UploadFile, File
8
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
9
  from sentence_transformers import SentenceTransformer
10
  from io import BytesIO
 
 
 
11
 
12
  # ---------------------------- #
13
  # CONFIGURATION
@@ -248,7 +251,7 @@ def faq(question: str):
248
  cleaned_answer = cleaned_answer.replace(token, "").strip()
249
 
250
  return {"answer": cleaned_answer}
251
-
252
 
253
  # --------- Gradio UI --------- #
254
  def gradio_upload(file):
 
8
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
9
  from sentence_transformers import SentenceTransformer
10
  from io import BytesIO
11
+ import gradio as gr
12
+ from gradio import mount_gradio_app
13
+
14
 
15
  # ---------------------------- #
16
  # CONFIGURATION
 
251
  cleaned_answer = cleaned_answer.replace(token, "").strip()
252
 
253
  return {"answer": cleaned_answer}
254
+
255
 
256
  # --------- Gradio UI --------- #
257
  def gradio_upload(file):