Update app.py

app.py

@@ -50,7 +50,11 @@ if uploaded_file:
     if st.button("Generate Style"):
         compiled_model, input_layer, output_layer = load_model(style)
         input_tensor = preprocess(image)
-        output = compiled_model([input_tensor])[output_layer]
+
+        # ✅ Correct inference syntax
+        result_dict = compiled_model.infer({input_layer: input_tensor})
+        output = result_dict[output_layer]
+
         result = postprocess(output)
         result_pil = Image.fromarray(result)
         col2.image(result_pil, caption=f"{style} Style", use_column_width=True)