Spaces:

yeq6x
/

PositionMapVisualizer

Sleeping

yeq6x commited on Jan 1

Commit

fab2b8a

1 Parent(s): 54507dc

load_model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ import torch
 from collections import Counter
 import random
-# import spaces
 pipe = None
@@ -136,7 +136,7 @@ def outpaint_image(image):
   return image
-# @spaces.GPU
 def predict_image(cond_image, prompt, negative_prompt):
     generator = torch.Generator()
     generator.manual_seed(random.randint(0, 2147483647))
@@ -160,7 +160,7 @@ def predict_image(cond_image, prompt, negative_prompt):
     return image
-# load_model()
 # Gradioアプリケーション
 with gr.Blocks() as demo:

 from collections import Counter
 import random
+import spaces
 pipe = None
   return image
+@spaces.GPU
 def predict_image(cond_image, prompt, negative_prompt):
     generator = torch.Generator()
     generator.manual_seed(random.randint(0, 2147483647))
     return image
+load_model()
 # Gradioアプリケーション
 with gr.Blocks() as demo: