Spaces:
Runtime error
Runtime error
Mariyaben
commited on
Commit
·
70a2285
1
Parent(s):
700a431
Add application file
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
3 |
|
4 |
# Replace with the actual model path or identifier
|
5 |
-
model_name = "meta-llama/Llama-
|
6 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
7 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
8 |
|
|
|
2 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
3 |
|
4 |
# Replace with the actual model path or identifier
|
5 |
+
model_name = "meta-llama/Llama-8B"
|
6 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
7 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
8 |
|