Indus 3.0 Demo

app.py

@@ -36,6 +36,15 @@ if os.path.exists(config_path):
 else:
     print(f"\nconfig.json not found at {config_path}")
 
+model_path = None
+for root, dirs, files in os.walk(base_dir):
+    if "config.json" in files:
+        model_path = root
+        break
+
+if model_path is None:
+    raise FileNotFoundError("config.json not found after extraction")
+
 # Load from the extracted folder
 tokenizer = AutoTokenizer.from_pretrained(extract_dir)
 model     = AutoModelForCausalLM.from_pretrained(extract_dir, torch_dtype="auto", device_map="auto")