Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -13,9 +13,10 @@ from transformers import pipeline
|
|
13 |
|
14 |
pipe = pipeline(
|
15 |
"text-generation",
|
16 |
-
model="
|
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"""
|