reichenbach commited on
Commit
01205ca
1 Parent(s): 6d073ef

Downloading Model Separately

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -7,6 +7,12 @@ from usellm import Message, Options, UseLLM
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
9
 
 
 
 
 
 
 
10
 
11
  def text_to_speech(text_input):
12
  CHUNK_SIZE = 1024
 
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
9
 
10
+ processor1 = WhisperProcessor.from_pretrained("openai/whisper-large-v2")
11
+ model1 = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large-v2")
12
+
13
+ tokenizer1 = AutoTokenizer.from_pretrained("microsoft/BioGPT-Large-PubMedQA", add_special_tokens=False)
14
+ model = AutoModelForCausalLM.from_pretrained("microsoft/BioGPT-Large-PubMedQA")#.to('cuda:0')
15
+
16
 
17
  def text_to_speech(text_input):
18
  CHUNK_SIZE = 1024