Spaces:
Sleeping
Sleeping
ansfarooq7
commited on
Commit
•
2ad4a2e
1
Parent(s):
8856002
Update app.py
Browse files
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")
|
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
|