Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Load the merged model
-model_name = "EmTpro01/merged-gemma"  # Replace with your merged model path
+model_name = "EmTpro01/gemma-paraphraser-4bit"  # Replace with your merged model path
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)  # Default device is CPU