Spaces:
Runtime error
Runtime error
Axolotlily
commited on
Commit
•
6142a36
1
Parent(s):
daa7e5a
Update app.py
Browse files
app.py
CHANGED
@@ -4,13 +4,13 @@ from aitextgen import aitextgen
|
|
4 |
ai = aitextgen(model="EleutherAI/gpt-neo-2.7B")
|
5 |
|
6 |
def ai_text(inp):
|
7 |
-
generated_text = ai.generate_one(max_length=
|
8 |
print(type(generated_text))
|
9 |
return generated_text
|
10 |
|
11 |
examples = [
|
12 |
["Artificial intelligence will"],
|
13 |
-
["Probably everyone
|
14 |
["Mars, Joyce and Venus had their first day at Blazing Inventions Academy in North Point, Hong Kong. The following day, while Mars, Joyce and Venus are eating breakfast in the Academy’s hall, a green gas spreads from North Point throughout the world."]
|
15 |
]
|
16 |
|
|
|
4 |
ai = aitextgen(model="EleutherAI/gpt-neo-2.7B")
|
5 |
|
6 |
def ai_text(inp):
|
7 |
+
generated_text = ai.generate_one(max_length=1000, prompt = inp, no_repeat_ngram_size=3)
|
8 |
print(type(generated_text))
|
9 |
return generated_text
|
10 |
|
11 |
examples = [
|
12 |
["Artificial intelligence will"],
|
13 |
+
["Probably everyone likes Turtles and Axolotls."],
|
14 |
["Mars, Joyce and Venus had their first day at Blazing Inventions Academy in North Point, Hong Kong. The following day, while Mars, Joyce and Venus are eating breakfast in the Academy’s hall, a green gas spreads from North Point throughout the world."]
|
15 |
]
|
16 |
|