Update app.py

app.py

@@ -1,19 +1,25 @@
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
 import gradio as gr
-from aitextgen import aitextgen
 
-ai = aitextgen(model="EleutherAI/gpt-neo-2.7B")
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
+model = AutoModelForCausalLM.from_pretrained("gpt2")
 
-def ai_text(inp):
-  generated_text = ai.generate_one(max_length=1001, prompt = inp, no_repeat_ngram_size=3)
-  print(type(generated_text))
+def text_generation(input_text, seed):
+  input_ids = tokenizer(input_text, return_tensors="pt").input_ids
+  torch.manual_seed(seed) # Max value: 18446744073709551615
+  outputs = model.generate(input_ids, do_sample=True, max_length=100)
+  generated_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)
   return generated_text
-  
-examples = [
-  ["Artificial intelligence will"],
-  ["Probably everyone likes Turtles and Axolotls."],
-  ["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."]
-   ] 
-  
-output_text = gr.outputs.Textbox()
-gr.Interface(ai_text,"textbox", output_text, title="Text Generator GPT-NEO-2.7B",
-             description="Copy or type text. Submit and the machine will generate text.", examples=examples).launch(share=False)
+
+title = "Text Generator Demo GPT2"
+description = "Text Generator Application by ecarbo"
+
+gr.Interface(
+    text_generation,
+    [gr.inputs.Textbox(lines=2, label="Enter input text"), gr.inputs.Number(default=10, label="Enter seed number")],
+    [gr.outputs.Textbox(type="auto", label="Text Generated")],
+    title=title,
+    description=description,
+    theme="huggingface"
+).launch()