Spaces:
Sleeping
Sleeping
syedMohib44
commited on
Commit
·
e90608d
1
Parent(s):
c5091e3
Add LFS
Browse files
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):
|