add imuru_large

app.py

@@ -28,6 +28,9 @@ model_zoo = {
     "imuru_small": {
         "repo_id": "Ruian7P/imuru_small",
     },
+    "imuru_large": {
+        "repo_id": "Ruian7P/imuru_large",
+    },
     # "emuru_t5_small": {
     #     "repo_id": "Ruian7P/emuru_result",
     #     "model_name": "emuru_t5_small_2e-5_ech5"
@@ -36,7 +39,7 @@ model_zoo = {
 
 model = None
 
-def load_model(model_name="imuru_small"):    
+def load_model(model_name="imuru_large"):    
     global model
 
     if model is None:
@@ -71,7 +74,7 @@ def process_image(img):
 
 
 @spaces.GPU
-def generate_handwriting(style_image, gen_text, model_name="imuru_small"):
+def generate_handwriting(style_image, gen_text, model_name="imuru_large"):
     """Generate handwriting in the style of the input image."""
     if not gen_text or gen_text.strip() == "":
         return None, "❌ Please provide text to generate"
@@ -144,7 +147,7 @@ with gr.Blocks(css=custom_css, title="Imuru") as demo:
             model_selector = gr.Dropdown(
                 label="🤖 Select Model",
                 choices=list(model_zoo.keys()),
-                value="imuru_small",
+                value="imuru_large",
                 interactive=True
             )