Karzan commited on
Commit
d7f1aae
1 Parent(s): 145884e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ from transformers import pipeline,GemmaForCausalLM,AutoTokenizer
2
  import gradio as gr
3
  import spaces
4
  # ignore_mismatched_sizes=True
5
- tokenizer = AutoTokenizer.from_pretrained('models/google/gemma-2-9b')
6
- model = GemmaForCausalLM.from_pretrained('models/google/gemma-2-9b',ignore_mismatched_sizes=True)
7
  pipe = pipeline('text-generation', model=model,tokenizer = tokenizer)
8
 
9
  @spaces.GPU(duration=120)
 
2
  import gradio as gr
3
  import spaces
4
  # ignore_mismatched_sizes=True
5
+ tokenizer = AutoTokenizer.from_pretrained('google/gemma-2-9b')
6
+ model = GemmaForCausalLM.from_pretrained('google/gemma-2-9b',ignore_mismatched_sizes=True)
7
  pipe = pipeline('text-generation', model=model,tokenizer = tokenizer)
8
 
9
  @spaces.GPU(duration=120)