old model

app.py

@@ -4,17 +4,17 @@ import numpy as np
 import gradio as gr
 import os
 
-#from model import ChaoticCoherentGenerator
-from model_256 import EfficientChaoticGenerator
+from model import ChaoticCoherentGenerator
+# from model_256 import EfficientChaoticGenerator
 from feature_extractor import CodeFeatureExtractor
 
 # ------------------- Device -------------------
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # ------------------- Load Model -------------------
-model = EfficientChaoticGenerator().to(device)
-checkpoint = torch.load("models/jid_model_256_e10.pth", map_location=device)
-model.load_state_dict(checkpoint["model_state_dict"])
+model = ChaoticCoherentGenerator().to(device)
+checkpoint = torch.load("models/chotic.pth", map_location=device)
+model.load_state_dict(checkpoint["state_dict"])
 model.eval()
 
 extractor = CodeFeatureExtractor()