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,
|