olivercareyncl commited on
Commit
dcef856
·
verified ·
1 Parent(s): 10d9251

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,9 +13,10 @@ from transformers import pipeline
13
 
14
  pipe = pipeline(
15
  "text-generation",
16
- model="HuggingFaceH4/zephyr-7b-beta" # <- Open-access model (no login needed)
17
  )
18
 
 
19
  # Generate a prompt for AI
20
  def generate_prompt(words):
21
  return f"""
 
13
 
14
  pipe = pipeline(
15
  "text-generation",
16
+ model="google/gemma-2b-it" # Smaller, free model that fits in memory
17
  )
18
 
19
+
20
  # Generate a prompt for AI
21
  def generate_prompt(words):
22
  return f"""