Spaces:
Runtime error
Runtime error
Commit
·
1e60c9b
1
Parent(s):
70de1fc
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
3 |
|
4 |
# Specify the path to your fine-tuned model and tokenizer
|
5 |
model_path = "./" # Assuming the model is in the same directory as your notebook
|
6 |
-
model_name = "pytorch_model-00001-of-00002.bin" # Replace with your model name
|
7 |
|
8 |
# Load the model and tokenizer
|
9 |
model = AutoModelForCausalLM.from_pretrained(model_path)
|
|
|
3 |
|
4 |
# Specify the path to your fine-tuned model and tokenizer
|
5 |
model_path = "./" # Assuming the model is in the same directory as your notebook
|
6 |
+
model_name = "https://huggingface.co/spaces/DR-Rakshitha/wizardlm_api/blob/main/pytorch_model-00001-of-00002.bin" # Replace with your model name
|
7 |
|
8 |
# Load the model and tokenizer
|
9 |
model = AutoModelForCausalLM.from_pretrained(model_path)
|