Update app.py

app.py

@@ -1,9 +1,7 @@
-# app.py
 import os
 import torch
 import gradio as gr
 from huggingface_hub import hf_hub_download, snapshot_download
-from importlib.util import spec_from_file_location, module_from_spec
 import sys
 
 def setup_cognitive_net():
@@ -11,9 +9,12 @@ def setup_cognitive_net():
     # Download repository content
     repo_path = snapshot_download(repo_id="VLabTech/cognitive_net")
     
-    # Import module from downloaded path
-    sys.path.append(repo_path)
-    from cognitive_net import DynamicCognitiveNet
+    # Add the repository path to Python path so we can import the package
+    if repo_path not in sys.path:
+        sys.path.insert(0, repo_path)
+    
+    # Import the package
+    from network import DynamicCognitiveNet
     return DynamicCognitiveNet
 
 def predict(input_text):