Spaces:

cyberpole
/

redornot

Sleeping

k-r-l commited on Jan 17

Commit

e652712

verified ·

1 Parent(s): c36de06

Update src/streamlit_app.py

Files changed (1) hide show

src/streamlit_app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, pipeline
 from peft import PeftModel
 from transformers import AutoModelForSequenceClassification
-LORA_MODEL = "cyberpole/gemma-3-270m-it-peft-SFTreward"
 BASE_MODEL = "google/gemma-3-270m-it"
 tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)

 from peft import PeftModel
 from transformers import AutoModelForSequenceClassification
+LORA_MODEL = "cyberpole/gemma-3-270m-it-peft-SFTreward_v2"
 BASE_MODEL = "google/gemma-3-270m-it"
 tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)