Update app.py

app.py

@@ -56,9 +56,14 @@ def text2story(text):
         # Generate the story - limit to a smaller max_length due to model size
         story = generator(prompt,
                          min_length=100,
+                         max_length=800,  
                          num_return_sequences=1,
-                         temperature=0.7,
-                         repetition_penalty=1.2)[0]['generated_text']
+                         top_k=50,  
+                         top_p=0.92, 
+                         no_repeat_ngram_size=3, 
+                         temperature=0.8,
+                         repetition_penalty=1.3,
+                         do_sample=True)[0]['generated_text']
         
         # Clean up the story by removing the prompt
         story = story.replace(prompt, "").strip()
@@ -138,7 +143,7 @@ async def text2audio_english(text):
 
 
 # Apply custom CSS for modern, stylish kid-friendly UI
-st.set_page_config(page_title="故事魔法", page_icon="✨", layout="wide")
+st.set_page_config(page_title="歡迎嚟到 ISOM 5240 - 故事魔法師！", page_icon="✨", layout="wide")
 
 load_css('styles.css')
 
@@ -150,9 +155,9 @@ progress_placeholder = st.empty()
 
 # File uploader with Cantonese
 with st.container():
-    st.subheader("揀一張靚相啦！")
+    st.subheader("請選擇一張畫作上載，格式必須係 JPG、JPEG 或 PNG！ 最大 200 MB !")
     uploaded_file = st.file_uploader(
-        "請上載一張圖片（只支援 JPG、JPEG 或 PNG 格式）",
+        "",
         type=["jpg", "jpeg", "png"],  # Limit file types
         key="upload"
     )
@@ -160,7 +165,7 @@ with st.container():
     if uploaded_file is not None:
         st.success("上載成功！")
     else:
-        st.info("請選擇一張靚相上載，格式必須係 JPG、JPEG 或 PNG！")
+        st.info("請選擇一張畫作上載，格式必須係 JPG、JPEG 或 PNG！ 最大 200 MB !")
         
 if uploaded_file is not None:
     # Save uploaded file
@@ -176,7 +181,7 @@ if uploaded_file is not None:
     if st.session_state.scenario is None:
         # Stage 1: Image to Text
         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("正在分析圖片...")