Update app.py

app.py

@@ -2,7 +2,10 @@ from transformers import pipeline
 import streamlit as st
 
 # Load the Gemma-7b text-generation pipeline
-generator = pipeline("text-generation", model="google/gemma-7b")
+# generator = pipeline("text-generation", model="google/gemma-7b")
+# Load model directly
+from transformers import AutoModel
+generator = AutoModel.from_pretrained("TheBloke/WhiteRabbitNeo-13B-GGUF")
 
 st.title("Project Prompt Generator")