ronniet commited on
Commit
253c97c
·
1 Parent(s): 3200948

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -15,8 +15,11 @@ tts_processor = SpeechT5Processor.from_pretrained(checkpoint)
15
  tts_model = SpeechT5ForTextToSpeech.from_pretrained(checkpoint)
16
  vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan")
17
 
18
- ic_processor = AutoProcessor.from_pretrained("microsoft/git-base")
19
- ic_model = AutoModelForCausalLM.from_pretrained("microsoft/git-base")
 
 
 
20
 
21
  def tts(text):
22
  if len(text.strip()) == 0:
 
15
  tts_model = SpeechT5ForTextToSpeech.from_pretrained(checkpoint)
16
  vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan")
17
 
18
+ # ic_processor = AutoProcessor.from_pretrained("microsoft/git-base")
19
+ # ic_model = AutoModelForCausalLM.from_pretrained("microsoft/git-base")
20
+
21
+ ic_processor = AutoProcessor.from_pretrained("ronniet/git-base-env")
22
+ ic_model = AutoModelForCausalLM.from_pretrained("ronniet/git-base-env")
23
 
24
  def tts(text):
25
  if len(text.strip()) == 0: