Update app.py

app.py

@@ -14,16 +14,15 @@ os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
 # Create the gr.State component *outside* the gr.Blocks context
 
 #predictor = gr.State(None)
-
 def init_predictor(task_type: str):
-    import torch
     from skyreelsinfer import TaskType
     from skyreelsinfer.offload import OffloadConfig
     from skyreelsinfer.skyreels_video_infer import SkyReelsVideoInfer
     from huggingface_hub.utils import RepositoryNotFoundError, RevisionNotFoundError, EntryNotFoundError
-#try:
-    predictor = SkyReelsVideoInfer(
-            task_type=TaskType.I2V, # if task_type == "i2v" else TaskType.T2V,
+
+    try:
+        predictor = SkyReelsVideoInfer(
+            task_type=TaskType.I2V if task_type == "i2v" else TaskType.T2V,
             model_id="Skywork/skyreels-v1-Hunyuan-i2v",
             quant_model=True,
             is_offload=True,
@@ -32,14 +31,13 @@ def init_predictor(task_type: str):
                 parameters_level=True,
             ),
             use_multiprocessing=False,
-    )
-    #return "Model loaded successfully!", predictor  # Return predictor
-    return predictor
+        )
+        return "Model loaded successfully!", predictor  # Return predictor
 
-#except (RepositoryNotFoundError, RevisionNotFoundError, EntryNotFoundError) as e:
-#    return f"Error: Model not found. Details: {e}", None
-#except Exception as e:
-#    return f"Error loading model: {e}", None
+    except (RepositoryNotFoundError, RevisionNotFoundError, EntryNotFoundError) as e:
+        return f"Error: Model not found. Details: {e}", None
+    except Exception as e:
+        return f"Error loading model: {e}", None
         
 predictor = init_predictor('i2v')