Spaces:

saadkiet
/

AI_Blog_generation_Powered_by_GPT_NEO_1.3B

Runtime error

saadkiet commited on Sep 10, 2022

Commit

45e95e0

•

1 Parent(s): 1f0211f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ from aitextgen import aitextgen
 import gradio as gr
-ai=aitextgen(model='EleutherAI/gpt-neo-2.7B',to_gpu=False) # EleutherAI/gpt-neo-2.7B EleutherAI/gpt-neo-1.3B
 def ai_text(Input):
   generated_text = ai.generate_one(max_length = 500, prompt = Input, no_repeat_ngram_size = 3) #repetition_penalty = 1.9)

 import gradio as gr
+ai=aitextgen(model='EleutherAI/gpt-neo-1.3B',to_gpu=False) # EleutherAI/gpt-neo-2.7B EleutherAI/gpt-neo-1.3B
 def ai_text(Input):
   generated_text = ai.generate_one(max_length = 500, prompt = Input, no_repeat_ngram_size = 3) #repetition_penalty = 1.9)