Update app.py

app.py

@@ -52,8 +52,8 @@ def text2story(text):
         inputs = tokenizer(prompt, return_tensors="pt")
         outputs = model.generate(
             inputs.input_ids,
-            min_length=100,
-            max_new_tokens=300,
+            min_length=50,
+            max_new_tokens=100,
             temperature=0.7,
             top_p=0.9,
             top_k=40,
@@ -71,8 +71,8 @@ def text2story(text):
         inputs = tokenizer(fallback_prompt, return_tensors="pt")
         fallback_story = fallback_generator.generate(
             inputs.input_ids,
-            min_length=100,
-            max_new_tokens=300,
+            min_length=50,
+            max_new_tokens=100,
             temperature=0.7,
             top_p=0.9,
             top_k=40,
@@ -204,7 +204,7 @@ if uploaded_file is not None:
 
         # Stage 2: Text to Story
         with st.container():
-            st.markdown("<h3><span class='stage-icon'>📝</span> 以下係我哋嘅魔術師覺得你嘅畫作係關於 : </h3>", unsafe_allow_html=True)
+            st.markdown("<h3><span class='stage-icon'>📝</span> 以下係我哋嘅魔術師正根據你嘅畫作嘅故事: </h3>", unsafe_allow_html=True)
             
             with progress_placeholder.container():
                 st.write("正在創作故事...")