azizmma commited on
Commit
405417e
1 Parent(s): 05c8259

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,14 +43,14 @@ num_return_sequences = st.sidebar.slider("Num Return Sequences", min_value = 1,
43
  num_beams = st.sidebar.slider("Num Beams", min_value = 2, max_value=6, value = 4)
44
  top_k = st.sidebar.slider("Top-k", min_value = 0, max_value=100, value = 90)
45
  top_p = st.sidebar.slider("Top-p", min_value = 0.4, max_value=1.0, step = 0.05, value = 0.9)
46
- repetition_penalty = st.sidebar.slider("Repetition Penalty", min_value = 0.45 max_value=2.0, step = 0.1, value = 1.2)
47
 
48
 
49
  if len(sent)<10:
50
  st.write("Input prompt is too small to generate")
51
  else:
52
  print(sent)
53
-
54
  output_sequences = generate_texts(pipelines[model_index],
55
  sent,
56
  max_length=max_length,
 
43
  num_beams = st.sidebar.slider("Num Beams", min_value = 2, max_value=6, value = 4)
44
  top_k = st.sidebar.slider("Top-k", min_value = 0, max_value=100, value = 90)
45
  top_p = st.sidebar.slider("Top-p", min_value = 0.4, max_value=1.0, step = 0.05, value = 0.9)
46
+ repetition_penalty = st.sidebar.slider("Repetition-Penalty", min_value = 0.45, max_value=2.0, step = 0.1, value = 1.2)
47
 
48
 
49
  if len(sent)<10:
50
  st.write("Input prompt is too small to generate")
51
  else:
52
  print(sent)
53
+ st.write(f"Generating for prompt {sent}....")
54
  output_sequences = generate_texts(pipelines[model_index],
55
  sent,
56
  max_length=max_length,