Spaces:
Running
Running
Ezi Ozoani
commited on
Commit
·
7f7f5a4
1
Parent(s):
fbdf912
task update
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ def img_to_bytes(img_path):
|
|
| 32 |
# sidebar
|
| 33 |
|
| 34 |
def load_model():
|
| 35 |
-
model_out = pipeline(task="
|
| 36 |
return model_out
|
| 37 |
|
| 38 |
def cs_sidebar():
|
|
|
|
| 32 |
# sidebar
|
| 33 |
|
| 34 |
def load_model():
|
| 35 |
+
model_out = pipeline(task="text-generation", model="distilgpt2")
|
| 36 |
return model_out
|
| 37 |
|
| 38 |
def cs_sidebar():
|