samidh commited on
Commit
d77692a
·
verified ·
1 Parent(s): 174f794

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -1,4 +1,9 @@
1
  import gradio as gr
2
  import os
3
 
4
- gr.load("models/samidh/cope-gemma2b-hs-2c-skr-s1.5.9.d25", hf_token=os.environ['HF_TOKEN']).launch()
 
 
 
 
 
 
1
  import gradio as gr
2
  import os
3
 
4
+ from transformers import AutoModelForCausalLM, AutoTokenizer
5
+
6
+ peft_model_id = "samidh/samidh-cope-gemma2b-hs-2c-skr-s1.5.9.d25"
7
+ model = AutoModelForCausalLM.from_pretrained(peft_model_id)
8
+
9
+ #gr.load("models/samidh/cope-gemma2b-hs-2c-skr-s1.5.9.d25", hf_token=os.environ['HF_TOKEN']).launch()