longer gpu

app.py

@@ -384,7 +384,7 @@ with gr.Blocks(theme="bethecloud/storj_theme", css=css) as demo:
     def turn_buttons_on():
         return gr.update(visible=True), gr.update(visible=True)
 
-@spaces.GPU(duration=8)
+@spaces.GPU(duration=59)
 @torch.inference_mode()
 def process_image(
     image,
@@ -433,11 +433,11 @@ def process_image(
         generator=generator,
         image=control_image,
     ).images[0]
-    torch.cuda.synchronize()
-    torch.cuda.empty_cache()
     print(f"\n-------------------------Preprocess done in: {preprocess_time:.2f} seconds-------------------------")    
     print(f"\n-------------------------Inference done in: {time.time() - start:.2f} seconds-------------------------")
     results.save("temp_image.jpg")
+    torch.cuda.synchronize()
+    torch.cuda.empty_cache()
     return results
 
 if prod:

local_app.py

@@ -431,38 +431,14 @@ def process_image(
         generator=generator,
         image=control_image,
     ).images[0]
-    torch.cuda.synchronize()
-    torch.cuda.empty_cache()
     print(f"\n-------------------------Preprocess done in: {preprocess_time:.2f} seconds-------------------------")    
     print(f"\n-------------------------Inference done in: {time.time() - start:.2f} seconds-------------------------")
-    
-    # timestamp = int(time.time())
-    #if not os.path.exists("./outputs"):
-    #    os.makedirs("./outputs")
-    # img_path = f"./{timestamp}.jpg"
-    # results_path = f"./{timestamp}_out_{prompt}.jpg"
-    # imageio.imsave(img_path, image)
-    # results.save(results_path)
     results.save("temp_image.jpg")
-    
-    # api.upload_file(
-    #     path_or_fileobj=img_path,
-    #     path_in_repo=img_path,
-    #     repo_id="broyang/anime-ai-outputs",
-    #     repo_type="dataset",
-    #     token=API_KEY,
-    #     run_as_future=True,
-    # )
-    # api.upload_file(
-    #     path_or_fileobj=results_path,
-    #     path_in_repo=results_path,
-    #     repo_id="broyang/anime-ai-outputs",
-    #     repo_type="dataset",
-    #     token=API_KEY,
-    #     run_as_future=True,
-    # )
-
+    torch.cuda.synchronize()
+    torch.cuda.empty_cache()
+    results.save("temp_image.jpg")
     return results
+
 if prod:
     demo.queue(max_size=20).launch(server_name="localhost", server_port=port)
 else: