werf

app.py

@@ -11,12 +11,11 @@ def sketch_transform(size):
     
 def generate_image(image):
     if image is not None:
-        sketch = image #['layers'][0]
+        sketch = image['layers'][0]
         sketch = transforms.ToTensor()(sketch).unsqueeze(0)
         sketch = sketch_transform(size=(256,256))(sketch)
         sketch = sketch/255.
-        print(sketch.shape)
-        print(type(sketch))
+
         with torch.no_grad():  # Disable gradient calculation for inference
             prompt='Realistic Seattle aerial satellite top view image with high quality details with buildings and roads in Seattle that probably has the following objects and characteristics: houses neighborhood city urban modern'
             model_output = pipe(prompt, num_inference_steps=20, generator=torch.manual_seed(0), image=sketch)
@@ -41,19 +40,22 @@ pipe.enable_model_cpu_offload(device=device)
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.ImageEditor(sources=(), brush=gr.Brush(colors=["#ffb266",  #building
-                                                          "#4059ff", #parking
-                                                          "#66ff66", #grass
-                                                         #"#009900", #forest
-                                                          "#cce5ff", #water
-                                                          #"#c0c0c0", #path
-                                                          "#606060" #road
-                                                         ], color_mode="fixed"))
+        gr.ImageEditor(sources=(), 
+                       image_mode='RGB', 
+                       brush=gr.Brush(colors=["#ffb266",  #building
+                                              "#4059ff", #parking
+                                              "#66ff66", #grass
+                                              #"#009900", #forest
+                                              "#cce5ff", #water
+                                              #"#c0c0c0", #path
+                                              "#606060" #road
+                                              ], color_mode="fixed")
+                      )
         #gr.Sketchpad()
         #gr.Image(shape=(512, 512), source="canvas", tool="sketch", image_mode="RGB", invert_colors=False, brush_color="black"),
         #gr.Dropdown(choices=allowed_colors, label="Brush Color")    
     ],
-    outputs=gr.Image()
+    outputs="image"
 )
 
 iface.launch()