Update app.py

app.py

@@ -5,11 +5,14 @@ import torch
 import torchvision.transforms as T
 import gradio as gr
 
+# Import your model creation function
+from model import create_model  # ここで create_model を定義しているファイルを指定してください
+
 # Load parameters and model
 with open("parameters.json", "r") as f:
     parameters = json.load(f)
 
-model = create_model(parameters)  # your existing create_model function
+model = create_model(parameters)
 weights = torch.load("cxp_projection_rotation.pt", map_location="cpu")
 model.load_state_dict(weights)
 model.eval()