Spaces:
Sleeping
Sleeping
shionhonda
commited on
Commit
•
278124e
1
Parent(s):
bc78d8c
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ title = st.text_input("Title : ")
|
|
16 |
# {"role": "user", "content": title}
|
17 |
# ], tokenize=False, add_generation_prompt=True)
|
18 |
# inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
|
19 |
-
generate_ids = model.generate(inputs.input_ids, max_new_tokens=50, do_sample=True, temperature=0.5, top_k=50, top_p=0.95)
|
20 |
|
21 |
submit = st.button("Submit")
|
22 |
|
|
|
16 |
# {"role": "user", "content": title}
|
17 |
# ], tokenize=False, add_generation_prompt=True)
|
18 |
# inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
|
19 |
+
# generate_ids = model.generate(inputs.input_ids, max_new_tokens=50, do_sample=True, temperature=0.5, top_k=50, top_p=0.95)
|
20 |
|
21 |
submit = st.button("Submit")
|
22 |
|