Commit ·
897c7ba
1
Parent(s): 547390f
update file 012
Browse files
app.py
CHANGED
|
@@ -10,7 +10,8 @@ app = Flask(__name__)
|
|
| 10 |
|
| 11 |
# Load lightweight model for CPU
|
| 12 |
# generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B", device=-1)
|
| 13 |
-
generator = pipeline("text-generation", model="
|
|
|
|
| 14 |
|
| 15 |
# Prompt template
|
| 16 |
template = PromptTemplate(
|
|
|
|
| 10 |
|
| 11 |
# Load lightweight model for CPU
|
| 12 |
# generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B", device=-1)
|
| 13 |
+
generator = pipeline("text-generation", model="gpt2", device=-1)
|
| 14 |
+
|
| 15 |
|
| 16 |
# Prompt template
|
| 17 |
template = PromptTemplate(
|