evanperez commited on
Commit
832d2d5
1 Parent(s): 607b52b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,9 +22,9 @@ os.system("pip install -r requirements.txt")
22
  import torch
23
  from transformers import AutoModelForCausalLM, AutoTokenizer
24
 
25
- model = AutoModelForCausalLM.from_pretrained("Writer/palmyra-small")
26
 
27
- tokenizer = AutoTokenizer.from_pretrained("Writer/palmyra-small")
 
28
 
29
 
30
  st.set_page_config(page_title="Gemini RAG", layout="wide")
 
22
  import torch
23
  from transformers import AutoModelForCausalLM, AutoTokenizer
24
 
 
25
 
26
+ tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b")
27
+ model = AutoModelForCausalLM.from_pretrained("google/gemma-7b")
28
 
29
 
30
  st.set_page_config(page_title="Gemini RAG", layout="wide")