Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import random
|
|
3 |
|
4 |
import gradio as gr
|
5 |
import numpy as np
|
6 |
-
|
7 |
import torch
|
8 |
import torchvision.transforms.functional as TF
|
9 |
from diffusers import (
|
@@ -112,7 +112,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
112 |
|
113 |
|
114 |
def run(
|
115 |
-
image: PIL.Image,
|
116 |
prompt: str,
|
117 |
negative_prompt: str,
|
118 |
style_name: str = DEFAULT_STYLE_NAME,
|
@@ -122,7 +122,7 @@ def run(
|
|
122 |
adapter_conditioning_factor: float = 0.8,
|
123 |
seed: int = 0,
|
124 |
progress=gr.Progress(track_tqdm=True),
|
125 |
-
) -> PIL.Image:
|
126 |
print(type(image))
|
127 |
image = image.convert("RGB")
|
128 |
image = TF.to_tensor(image) > 0.5
|
|
|
3 |
|
4 |
import gradio as gr
|
5 |
import numpy as np
|
6 |
+
import PIL.Image
|
7 |
import torch
|
8 |
import torchvision.transforms.functional as TF
|
9 |
from diffusers import (
|
|
|
112 |
|
113 |
|
114 |
def run(
|
115 |
+
image: PIL.Image.Image,
|
116 |
prompt: str,
|
117 |
negative_prompt: str,
|
118 |
style_name: str = DEFAULT_STYLE_NAME,
|
|
|
122 |
adapter_conditioning_factor: float = 0.8,
|
123 |
seed: int = 0,
|
124 |
progress=gr.Progress(track_tqdm=True),
|
125 |
+
) -> PIL.Image.Image:
|
126 |
print(type(image))
|
127 |
image = image.convert("RGB")
|
128 |
image = TF.to_tensor(image) > 0.5
|