Update app.py

app.py

@@ -1,11 +1,17 @@
-from optimum.intel import OVPipelineForText2Image
-from openvino import Core, Model
+from optimum.intel import OVStableDiffusionPipeline  # Changed from OVPipelineForText2Image
+from openvino import Core, Model  # Updated to use non-deprecated openvino module
 import gradio as gr
 
 # Load the model
 model_id = "AIFunOver/FLUX.1-dev-openvino-fp16"
 try:
-    model = OVPipelineForText2Image.from_pretrained(model_id)
+    model = OVStableDiffusionPipeline.from_pretrained(
+        model_id,
+        export=True,  # Enable export to OpenVINO IR if needed
+        device="AUTO"  # Use AUTO for flexible device selection
+    )
+    # Save the model after export to avoid re-conversion
+    model.save_pretrained("./flux1_dev_openvino_fp16")
 except Exception as e:
     print(f"Error loading model: {e}")
     exit(1)
@@ -16,10 +22,10 @@ def generate_image(prompt, num_inference_steps=50, guidance_scale=7.5):
         # Generate image using the model
         result = model(
             prompt=prompt,
-            num_inference_steps=num_inference_steps,
-            guidance_scale=guidance_scale,
-            height=512,  # Adjust as needed
-            width=512    # Adjust as needed
+            num_inference_steps=int(num_inference_steps),  # Ensure integer input
+            guidance_scale=float(guidance_scale),  # Ensure float input
+            height=512,
+            width=512
         )
         image = result.images[0]  # Extract the generated image
         return image
@@ -41,4 +47,7 @@ interface = gr.Interface(
 
 # Launch the Gradio app
 if __name__ == "__main__":
-    interface.launch()
+    try:
+        interface.launch(server_name="0.0.0.0", server_port=7860)  # Explicit server settings
+    except Exception as e:
+        print(f"Error launching Gradio interface: {e}")