ManishThota commited on
Commit
393adda
1 Parent(s): b6fa2c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,11 +15,11 @@ from io import StringIO
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
  # Initialize the model and tokenizer
18
- model = AutoModelForCausalLM.from_pretrained("ManishThota/SparrowVQE",
19
  torch_dtype=torch.float16,
20
  device_map="auto",
21
  trust_remote_code=True)
22
- tokenizer = AutoTokenizer.from_pretrained("ManishThota/SparrowVQE", trust_remote_code=True)
23
 
24
 
25
 
 
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
  # Initialize the model and tokenizer
18
+ model = AutoModelForCausalLM.from_pretrained("ManishThota/CustomModel",
19
  torch_dtype=torch.float16,
20
  device_map="auto",
21
  trust_remote_code=True)
22
+ tokenizer = AutoTokenizer.from_pretrained("ManishThota/CustomModel", trust_remote_code=True)
23
 
24
 
25