Upload app.py with huggingface_hub

app.py

@@ -78,7 +78,7 @@ def load_model():
     return model
 
 @spaces.GPU
-def generate_handwriting(style_image, style_text, gen_text, cfg_scale=1.0, max_tokens=128):
+def generate_handwriting(style_image, style_text, gen_text, cfg_scale=1.5):
     """
     Generate handwriting based on style image and generation text
     
@@ -128,7 +128,7 @@ def generate_handwriting(style_image, style_text, gen_text, cfg_scale=1.0, max_t
                 gen_img=None,
                 style_len=style_len,
                 gen_len=None,
-                max_img_len=max_tokens * 8
+                max_img_len=128 * 8
             )
             
             # Generate
@@ -137,7 +137,7 @@ def generate_handwriting(style_image, style_text, gen_text, cfg_scale=1.0, max_t
                 style_text=[style_text if style_text else ""],
                 gen_text=[gen_text],
                 cfg_scale=cfg_scale,
-                max_new_tokens=max_tokens
+                max_new_tokens=128
             )
             
             # Convert to PIL Image
@@ -210,23 +210,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 value="Hello World!"
             )
             
-            with gr.Row():
-                cfg_scale = gr.Slider(
-                    minimum=1.0,
-                    maximum=5.0,
-                    value=1.5,
-                    step=0.1,
-                    label="CFG Scale",
-                    info="Higher values = stronger style guidance"
-                )
-                max_tokens = gr.Slider(
-                    minimum=32,
-                    maximum=256,
-                    value=128,
-                    step=32,
-                    label="Max Tokens",
-                    info="Maximum length of generated text"
-                )
+            cfg_scale = gr.Slider(
+                minimum=1.0,
+                maximum=5.0,
+                value=1.5,
+                step=0.1,
+                label="CFG Scale",
+                info="Higher values = stronger style guidance"
+            )
             
             generate_btn = gr.Button("Generate ✨", variant="primary", size="lg")
         
@@ -240,19 +231,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # Examples with sample images
     gr.Examples(
         examples=[
-            ["example_images/handwritten_1.png", "", "Hello from handwritten style!", 1.5, 128],
-            ["example_images/handwritten_2.png", "", "The quick brown fox jumps over the lazy dog", 2.0, 128],
-            ["example_images/typewritten_1.png", "", "This is typewritten style generation", 1.5, 128],
-            [None, "", "No style reference - default generation", 1.0, 96],
+            ["example_images/handwritten_1.png", "", "The quick brown fox jumps over the lazy dog", 1.5],
+            ["example_images/handwritten_2.png", "", "Hello from handwritten style!", 1.5],
+            ["example_images/handwritten_3.png", "", "Artificial Intelligence and Machine Learning", 2.0],
+            ["example_images/typewritten_1.png", "", "This is typewritten style generation", 1.5],
         ],
-        inputs=[style_image_input, style_text_input, gen_text_input, cfg_scale, max_tokens],
+        inputs=[style_image_input, style_text_input, gen_text_input, cfg_scale],
         label="Example Styles"
     )
     
     # Connect the generation function
     generate_btn.click(
         fn=generate_handwriting,
-        inputs=[style_image_input, style_text_input, gen_text_input, cfg_scale, max_tokens],
+        inputs=[style_image_input, style_text_input, gen_text_input, cfg_scale],
         outputs=[output_image, status_text]
     )