Spaces:
Sleeping
Sleeping
Asankhaya Sharma
commited on
Commit
·
b351fdf
1
Parent(s):
6f87c71
update app.py
Browse files
app.py
CHANGED
@@ -37,8 +37,6 @@ def submit():
|
|
37 |
# prompt = "How long will it take for the poc to finish?"
|
38 |
# inputs = tokenizer(prompt, return_tensors="pt")
|
39 |
|
40 |
-
max_tokens = st.sidebar.slider("Max Tokens", min_value = 16, max_value = 64, value = 32)
|
41 |
-
|
42 |
model, tokenizer = get_model()
|
43 |
generation_config = GenerationConfig(max_new_tokens=max_tokens,
|
44 |
num_beams=4,
|
|
|
37 |
# prompt = "How long will it take for the poc to finish?"
|
38 |
# inputs = tokenizer(prompt, return_tensors="pt")
|
39 |
|
|
|
|
|
40 |
model, tokenizer = get_model()
|
41 |
generation_config = GenerationConfig(max_new_tokens=max_tokens,
|
42 |
num_beams=4,
|