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):
|