hamxaameer commited on
Commit
1d12e3d
·
verified ·
1 Parent(s): a70f555

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import os
5
 
6
  # Model configuration
7
- MODEL_NAME = "your-username/your-model-name" # Replace with your actual HF model repo
8
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
  # Load model and tokenizer
 
4
  import os
5
 
6
  # Model configuration
7
+ MODEL_NAME = "model.safetensors" # Replace with your actual HF model repo
8
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
  # Load model and tokenizer