ajv009 commited on
Commit
e3db908
1 Parent(s): e6c914c

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ HF_TOKEN = os.environ.get("HF_TOKEN", None)
10
 
11
 
12
  # Load the tokenizer and model
13
- tokenizer = AutoTokenizer.from_pretrained("models/Telugu-LLM-Labs/Indic-gemma-7b-finetuned-sft-Navarasa-2.0")
14
- model = AutoModelForCausalLM.from_pretrained("models/Telugu-LLM-Labs/Indic-gemma-7b-finetuned-sft-Navarasa-2.0", device_map="auto")
15
 
16
 
17
  @spaces.GPU(duration=120)
 
10
 
11
 
12
  # Load the tokenizer and model
13
+ tokenizer = AutoTokenizer.from_pretrained("Telugu-LLM-Labs/Indic-gemma-7b-finetuned-sft-Navarasa-2.0")
14
+ model = AutoModelForCausalLM.from_pretrained("Telugu-LLM-Labs/Indic-gemma-7b-finetuned-sft-Navarasa-2.0", device_map="auto")
15
 
16
 
17
  @spaces.GPU(duration=120)