Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
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,
|