Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-
|
|
20 |
# Initialize the summarization model and tokenizer
|
21 |
# Use a smaller version of the Llama model and load in FP16
|
22 |
summarization_model = AutoModelForCausalLM.from_pretrained(
|
23 |
-
"meta-llama/
|
24 |
torch_dtype=torch.float16,
|
25 |
low_cpu_mem_usage=True
|
26 |
)
|
|
|
20 |
# Initialize the summarization model and tokenizer
|
21 |
# Use a smaller version of the Llama model and load in FP16
|
22 |
summarization_model = AutoModelForCausalLM.from_pretrained(
|
23 |
+
"meta-llama/LlamaGuard-7b",
|
24 |
torch_dtype=torch.float16,
|
25 |
low_cpu_mem_usage=True
|
26 |
)
|