Update app.py

app.py

@@ -13,13 +13,18 @@ import requests
 from loguru import logger
 from decouple import config
 
+from pathlib import Path
+from PIL import Image
+import io
+
 URL = config('URL')
 OUTPUT_DIR = config('OUTPUT_DIR')
+INPUT_DIR = config('INPUT_DIR')
 COMF_PATH = config('COMF_PATH')
 
 import torch
 
-import spaces
+#import spaces
 
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
@@ -50,10 +55,14 @@ def check_server_ready():
         return False
 
 
-@spaces.GPU
-def generate_image(prompt):
+#@spaces.GPU
+def generate_image(prompt, image):
     previous_image = get_latest_image(OUTPUT_DIR)
 
+    image = Image.fromarray(image)
+    image.save(OUTPUT_DIR+'/input.png', format='PNG')
+
+
     # Запускаем скрипт как подпроцесс
     process = subprocess.Popen([sys.executable, COMF_PATH, "--listen", "127.0.0.1"])
     logger.debug(f'Subprocess started with PID: {process.pid}')
@@ -97,7 +106,7 @@ def generate_image(prompt):
 
 
 if __name__ == "__main__":
-    demo = gr.Interface(fn=generate_image, inputs=["text"], outputs=["image"])
+    demo = gr.Interface(fn=generate_image, inputs=["text", "image"], outputs=["image"])
     demo.launch(debug=True)
     logger.debug('demo.launch()')