Update app.py

app.py

@@ -9,17 +9,21 @@ model_id = "None1145/noobai-XL-Vpred-0.65s-openvino"
 pipe = OVStableDiffusionXLPipeline.from_pretrained(model_id)
 
 # Initialize previous dimensions
-prev_height = None
-prev_width = None
+prev_height = 1216
+prev_width = 832
+pipe.reshape(batch_size=1, height=prev_height, width=prev_width, num_images_per_prompt=1)
+pipe.compile()
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
 def reload_model(new_model_id):
-    global pipe, model_id
+    global pipe, model_id, prev_height, prev_width
     model_id = new_model_id
     try:
         pipe = OVStableDiffusionXLPipeline.from_pretrained(model_id)
+        pipe.reshape(batch_size=1, height=prev_height, width=prev_width, num_images_per_prompt=1)
+        pipe.compile()
         return f"Model successfully loaded: {model_id}"
     except Exception as e:
         return f"Failed to load model: {str(e)}"
@@ -141,7 +145,7 @@ with gr.Blocks() as img:
     
     gr.Markdown("### Model Reload")
     with gr.Row():
-        new_model_id = gr.Text(label="New Model ID", placeholder="Enter new model ID")
+        new_model_id = gr.Text(label="New Model ID", placeholder="Enter model ID", value=model_id)
         reload_button = gr.Button("Reload Model", variant="primary")
         reload_status = gr.Text(label="Status", interactive=False)