olivercareyncl commited on
Commit
369ac08
·
verified ·
1 Parent(s): dcef856

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,10 +13,11 @@ from transformers import pipeline
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"""
 
13
 
14
  pipe = pipeline(
15
  "text-generation",
16
+ model="tiiuae/falcon-1b" # ✅ 1B Model - Fully Open, No Login Needed
17
  )
18
 
19
 
20
+
21
  # Generate a prompt for AI
22
  def generate_prompt(words):
23
  return f"""