pratikshahp commited on
Commit
7ff7916
·
verified ·
1 Parent(s): 39357c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -5,9 +5,10 @@ from dotenv import load_dotenv
5
  import os
6
  load_dotenv()
7
 
8
- tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
9
- model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
10
-
 
11
  #hf_token=os.getenv("HF_TOKEN")
12
 
13
  # Load the model and tokenizer
 
5
  import os
6
  load_dotenv()
7
 
8
+ #tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
9
+ #model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
10
+ tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b")
11
+ model = AutoModelForCausalLM.from_pretrained("google/gemma-2b")
12
  #hf_token=os.getenv("HF_TOKEN")
13
 
14
  # Load the model and tokenizer