amazonaws-sp commited on
Commit
50611af
·
verified ·
1 Parent(s): 944fca9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -2,15 +2,17 @@
2
 
3
  from __future__ import annotations
4
 
 
5
  import os
6
  import random
7
 
8
  import gradio as gr
9
  import numpy as np
10
- import PIL.Image
11
  import spaces
12
  import torch
13
- from diffusers import AutoencoderKL, DiffusionPipeline
 
 
14
 
15
  DESCRIPTION = "# Run any LoRA or SD Model"
16
  if not torch.cuda.is_available():
@@ -52,7 +54,6 @@ def generate(
52
  vaecall = 'madebyollin/sdxl-vae-fp16-fix',
53
  lora = '',
54
  lora_scale: float = 0.7,
55
- ) -> PIL.Image.Image:
56
  if torch.cuda.is_available():
57
 
58
  if not use_vae:
 
2
 
3
  from __future__ import annotations
4
 
5
+ import requests
6
  import os
7
  import random
8
 
9
  import gradio as gr
10
  import numpy as np
 
11
  import spaces
12
  import torch
13
+ from PIL import Image
14
+ from io import BytesIO
15
+ from diffusers import AutoencoderKL, DiffusionPipeline, StableDiffusionImg2ImgPipeline
16
 
17
  DESCRIPTION = "# Run any LoRA or SD Model"
18
  if not torch.cuda.is_available():
 
54
  vaecall = 'madebyollin/sdxl-vae-fp16-fix',
55
  lora = '',
56
  lora_scale: float = 0.7,
 
57
  if torch.cuda.is_available():
58
 
59
  if not use_vae: