Update app.py

app.py

@@ -4,12 +4,22 @@ from os import system
 system("pip3 install torch")
 system("pip3 install transformers")
 
-from transformers import AutoTokenizer
+from transformers import AutoTokenizer,OpenAIGPTLMHeadModel
 
 tokenizer = AutoTokenizer.from_pretrained("openai-community/openai-gpt")
+model = OpenAIGPTLMHeadModel.from_pretrained("openai-community/openai-gpt")
 
 def work(inp_text):
-    out = tokenizer.encode(inp_text)
+    out = tokenizer.encode(inp_text,return_tensors="pt")
+
+    out = model.generate(
+        out,
+        max_new_tokens=20,
+        do_sample=True,
+        temperature=0.3,
+    )
+
+    out = tokenizer.decode(out[0])
 
     return str(out)