Spaces:

raygiles3
/

speech_analyzer

Runtime error

raygiles3 commited on Jul 19, 2024

Commit

f5f8752

•

1 Parent(s): 5a9c558

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ summarization_model = AutoModelForCausalLM.from_pretrained(
     torch_dtype=torch.float16,
     low_cpu_mem_usage=True
 )
-summarization_tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-3b-hf")
 # Function to transcribe audio
 def transcribe_audio(audio_file):

     torch_dtype=torch.float16,
     low_cpu_mem_usage=True
 )
+summarization_tokenizer = AutoTokenizer.from_pretrained("meta-llama/LlamaGuard-7b")
 # Function to transcribe audio
 def transcribe_audio(audio_file):