ansfarooq7 commited on
Commit
2ad4a2e
1 Parent(s): 8856002

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -129,7 +129,7 @@ def get_rhyming_line(prompt, rhyming_word, inputs_len):
129
  return final_sentence
130
 
131
  def gptneo_summary(topic):
132
- input_ids = gptneo_tokenizer(f"Here is some information about {topic}", return_tensors="pt").input_ids.to(device)
133
  generated_ids = gptneo_model.generate(input_ids, do_sample=True, temperature=0.9, max_length=200)
134
  generated_text = gptneo_tokenizer.decode(generated_ids[0])
135
  return generated_text
 
129
  return final_sentence
130
 
131
  def gptneo_summary(topic):
132
+ input_ids = gptneo_tokenizer(f"Here is some information about {topic}", return_tensors="pt")
133
  generated_ids = gptneo_model.generate(input_ids, do_sample=True, temperature=0.9, max_length=200)
134
  generated_text = gptneo_tokenizer.decode(generated_ids[0])
135
  return generated_text