Update app.py

app.py

@@ -129,9 +129,9 @@ def get_rhyming_line(prompt, rhyming_word, inputs_len):
     return final_sentence
 
 def gptneo_summary(topic):
-    input_ids = gptneo_tokenizer(f"Here is some information about {topic}", return_tensors="pt")
-    generated_ids = gptneo_model.generate(input_ids, do_sample=True, temperature=0.9, max_length=200)
-    generated_text = gptneo_tokenizer.decode(generated_ids[0])
+    input_ids = gptneo_tokenizer(f"Here is some information about {topic}", return_tensors="pt").input_ids
+    gen_tokens = gptneo_model.generate(input_ids, do_sample=True, temperature=0.9, max_length=200)
+    generated_text = gptneo_tokenizer.decode(gen_tokens[0])
     return generated_text 
     
 def generate(topic, wiki=True):