Update app.py

app.py

@@ -3,6 +3,8 @@ import shutil
 import gradio as gr
 from google.colab import files
 from PIL import Image
+import subprocess
+
 
 def process_images(cloth_image, origin_image):
     # Convert numpy arrays to PIL images
@@ -23,7 +25,8 @@ def process_images(cloth_image, origin_image):
     print("saved")
     # Run your main.py script (assuming it's in the same directory)
     # Note: You might need to adjust the command based on your actual script's requirements
-    !python main.py
+    command = ["python", "main.py"]
+    subprocess.run(command, check=True)
     print("might run")
     # Return the final image
     final_image_path = os.path.join(input_dir, 'finalimg.png')