Update app.py

app.py

@@ -133,9 +133,7 @@ def get_rhyming_line(prompt, rhyming_word, inputs_len):
     return final_sentence
 
 def gptj_summary(topic):
-    input_ids = gptj_tokenizer(f"Here is some information about {topic}", return_tensors="pt").input_ids
-
-    input_ids = gptj_tokenizer(prompt, return_tensors="pt").input_ids.to(device)
+    input_ids = gptj_tokenizer(f"Here is some information about {topic}", return_tensors="pt").input_ids.to(device)
     generated_ids = gptj_model.generate(input_ids, do_sample=True, temperature=0.9, max_length=200)
     generated_text = gptj_tokenizer.decode(generated_ids[0])
     return generated_text