Asankhaya Sharma commited on
Commit
b351fdf
·
1 Parent(s): 6f87c71

update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
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,