sahilmayekar commited on
Commit
13e1200
·
verified ·
1 Parent(s): cb066d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ ds = load_dataset("Vezora/Open-Critic-GPT")
14
  st.write("Dataset")
15
 
16
  # Load the model and tokenizer
17
- model_name = "meta-llama/Meta-Llama-3-8B"
18
  model = AutoModelForCausalLM.from_pretrained(model_name)
19
  tokenizer = AutoTokenizer.from_pretrained(model_name)
20
 
 
14
  st.write("Dataset")
15
 
16
  # Load the model and tokenizer
17
+ model_name = "google/gemma-2-2b"
18
  model = AutoModelForCausalLM.from_pretrained(model_name)
19
  tokenizer = AutoTokenizer.from_pretrained(model_name)
20