Spaces:

Wootang01
/

text_generator_two

Runtime error

Wootang01 commited on Mar 4, 2022

Commit

78ef565

1 Parent(s): b36c880

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from aitextgen import aitextgen
-ai = aitextgen(model="EleutherAI/gpt-neo-2.7B")
 def ai_text(inp):
   generated_text = ai.generate_one(max_length=85, prompt = inp, no_repeat_ngram_size=3)

 import gradio as gr
 from aitextgen import aitextgen
+ai = aitextgen(model="hivemind/gpt-j-6B-8bit")
 def ai_text(inp):
   generated_text = ai.generate_one(max_length=85, prompt = inp, no_repeat_ngram_size=3)