should work this time

app.py

@@ -25,19 +25,19 @@ ensure_env_installed()
 
 # Global model variable
 model_zoo = {
-    "imuru_t5_small": {
-        "repo_id": "Ruian7P/emuru_result",
-        "model_name": "imuru_t5_small_2e-5_ech5"
+    "imuru_small": {
+        "repo_id": "Ruian7P/imuru_small",
+        "model_name": None
     },
-    "emuru_t5_small": {
-        "repo_id": "Ruian7P/emuru_result",
-        "model_name": "emuru_t5_small_2e-5_ech5"
-    }
+    # "emuru_t5_small": {
+    #     "repo_id": "Ruian7P/emuru_result",
+    #     "model_name": "emuru_t5_small_2e-5_ech5"
+    # }
 }
 
 model = None
 
-def load_model(model_name="imuru_t5_small"):    
+def load_model(model_name="imuru_small"):    
     global model
 
     if model is None:
@@ -45,7 +45,8 @@ def load_model(model_name="imuru_t5_small"):
         from transformers import AutoModel
         
         model = AutoModel.from_pretrained(
-            f"{model_zoo[model_name]['repo_id']}/{model_zoo[model_name]['model_name']}",
+            model_zoo[model_name]["repo_id"],
+            subfolder=model_zoo[model_name]["model_name"],
             trust_remote_code=True
         )
         model.eval()
@@ -72,7 +73,7 @@ def process_image(img):
 
 
 @spaces.GPU
-def generate_handwriting(style_image, gen_text, model_name="imuru_t5_small"):
+def generate_handwriting(style_image, gen_text, model_name="imuru_small"):
     """Generate handwriting in the style of the input image."""
     if not gen_text or gen_text.strip() == "":
         return None, "❌ Please provide text to generate"
@@ -143,7 +144,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_t5_small",
+                value="imuru_small",
                 interactive=True
             )