Update app.py
Browse files
app.py
CHANGED
|
@@ -6,6 +6,10 @@ from transformers import pipeline
|
|
| 6 |
|
| 7 |
from share_btn import community_icon_html, loading_icon_html, share_js
|
| 8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
# whisper model specification
|
| 10 |
model = whisper.load_model("tiny")
|
| 11 |
|
|
|
|
| 6 |
|
| 7 |
from share_btn import community_icon_html, loading_icon_html, share_js
|
| 8 |
|
| 9 |
+
# get gpt2 model
|
| 10 |
+
generator = pipeline('text-generation', model='gpt2')
|
| 11 |
+
|
| 12 |
+
|
| 13 |
# whisper model specification
|
| 14 |
model = whisper.load_model("tiny")
|
| 15 |
|