Spaces:
Runtime error
Runtime error
Commit
·
a36c788
1
Parent(s):
9194ef4
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
from aitextgen import aitextgen
|
| 3 |
# Downloading gpt neo from hugging face model hub
|
| 4 |
-
ai= aitextgen(model='EleutherAI/gpt-neo-125M', to_gpu=
|
| 5 |
def ai_text(input):
|
| 6 |
# returning text as string for gradio
|
| 7 |
generated_text= ai.generate_one(max_length=1000, prompt=input, no_repeat_ngram_size=3)
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
from aitextgen import aitextgen
|
| 3 |
# Downloading gpt neo from hugging face model hub
|
| 4 |
+
ai= aitextgen(model='EleutherAI/gpt-neo-125M', to_gpu=False)
|
| 5 |
def ai_text(input):
|
| 6 |
# returning text as string for gradio
|
| 7 |
generated_text= ai.generate_one(max_length=1000, prompt=input, no_repeat_ngram_size=3)
|