v12

app.py

@@ -11,6 +11,7 @@ def test_v01():
     top_k = st.sidebar.slider("Top-k", min_value = 0, max_value=5, value = 0)
     top_p = st.sidebar.slider("Top-p", min_value = 0.0, max_value=1.0, step = 0.05, value = 0.9)
     num_return_sequences = st.sidebar.number_input('Number of Return Sequences', min_value=1, max_value=5, value=1, step=1)
+    return num_return_sequences
 
 def test_v02():
     dataset = load_dataset("merve/poetry", streaming=True)
@@ -21,5 +22,6 @@ def test_v02():
     sns.catplot(x="author", data=df, kind="count", aspect = 4)
     plt.xticks(rotation=90)
     st.pyplot()
+    return st
 
-test_v01()
+test_v02()