Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
import time
|
3 |
-
from transformers import pipeline,
|
4 |
from sentence_transformers import SentenceTransformer
|
5 |
from sklearn.metrics.pairwise import cosine_similarity
|
6 |
from TTS.api import TTS # Coqui TTS library
|
@@ -12,7 +12,7 @@ embedding_model = SentenceTransformer("all-MiniLM-L6-v2")
|
|
12 |
|
13 |
gpt_model_name = "google/flan-t5-base"
|
14 |
gpt_tokenizer = AutoTokenizer.from_pretrained(gpt_model_name)
|
15 |
-
gpt_model =
|
16 |
|
17 |
tts_model = TTS(model_name="tts_models/en/ljspeech/tacotron2-DDC", progress_bar=False, gpu=False)
|
18 |
|
|
|
1 |
import gradio as gr
|
2 |
import time
|
3 |
+
from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
|
4 |
from sentence_transformers import SentenceTransformer
|
5 |
from sklearn.metrics.pairwise import cosine_similarity
|
6 |
from TTS.api import TTS # Coqui TTS library
|
|
|
12 |
|
13 |
gpt_model_name = "google/flan-t5-base"
|
14 |
gpt_tokenizer = AutoTokenizer.from_pretrained(gpt_model_name)
|
15 |
+
gpt_model = AutoModelForSeq2SeqLM.from_pretrained(gpt_model_name)
|
16 |
|
17 |
tts_model = TTS(model_name="tts_models/en/ljspeech/tacotron2-DDC", progress_bar=False, gpu=False)
|
18 |
|