Upload app.py

app.py

@@ -221,9 +221,9 @@ def build_textured_cube(textures, out_dir=None):
 # Fonction appelée par Gradio
 # retourne : 4 miniatures (PIL) + chemin vers le .obj (str)
 # -------------------------
-def run(img1, img2):
+def run(fibers: Image.Image, rings: Image.Image, num_steps: int):
     try:
-        outputs = inference(img1, img2, 10)
+        outputs = inference(fibers, rings, int(num_steps))
         if not (isinstance(outputs, (list, tuple)) and len(outputs) >= 4):
             raise ValueError("user_inference doit renvoyer une liste/tuple de 4 images.")
         # Prendre les 4 premières images et convertir en PIL RGB
@@ -258,8 +258,8 @@ with gr.Blocks(title="Cube 3D — coller 4 images sur les faces") as demo:
     gr.Markdown("**Chargez 2 images** → votre code d'inference doit produire 4 images → les 4 images seront collées sur les faces latérales d'un cube (.obj).")
     with gr.Row():
         with gr.Column(scale=1):
-            inp1 = gr.Image(label="Image 1 (entrée)", type="pil")
-            inp2 = gr.Image(label="Image 2 (entrée)", type="pil")
+            inp1 = gr.Image(label="Image 1 (entrée)", type="numpy")
+            inp2 = gr.Image(label="Image 2 (entrée)", type="numpy")
             run_btn = gr.Button("Lancer l'inference")
             gr.Markdown("Remplace `user_inference` dans `app.py` par ta fonction d'inference réelle.")
         with gr.Column(scale=2):