Spaces:

arnab12345678
/

grammar-assistant

Runtime error

arnab12345678 commited on Mar 22, 2025

Commit

dfeeed5

verified ·

1 Parent(s): 1b82705

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 # Load Model and Tokenizer
-model_name = "your-username/grammar-assistant"  # Replace with your model repo
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = AutoModelForCausalLM.from_pretrained(model_name).to(device)

 from transformers import AutoModelForCausalLM, AutoTokenizer
 # Load Model and Tokenizer
+model_name = "arnab12345678grammar-assistant"  # Replace with your model repo
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = AutoModelForCausalLM.from_pretrained(model_name).to(device)