garyd1 commited on
Commit
13da39a
·
verified ·
1 Parent(s): 4faaac7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import time
3
- from transformers import pipeline, AutoModelForCausalLM, 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,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 = AutoModelForCausalLM.from_pretrained(gpt_model_name)
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