Spaces:

crystal99
/

My-First-Fine-Tuned-Model

Sleeping

crystal99 commited on Jan 27

Commit

2395396

verified ·

1 Parent(s): c7fbf53

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 # Load your fine-tuned model and tokenizer
-model_name = "openai-community/gpt2" #crystal99/my-fine-tuned-model
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 import torch
 # Load your fine-tuned model and tokenizer
+model_name = "crystal99/my-fine-tuned-model" #crystal99/my-fine-tuned-model
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)