vcollos commited on
Commit
194b5c6
·
verified ·
1 Parent(s): 0117aaf

mais testes

Browse files
Files changed (1) hide show
  1. app.py +10 -5
app.py CHANGED
@@ -6,10 +6,17 @@ import random
6
  import logging
7
  import numpy as np
8
  from typing import Any, Dict, List, Optional, Union
9
-
 
10
  import torch
11
  from PIL import Image
12
- import gradio as gr
 
 
 
 
 
 
13
 
14
 
15
  from diffusers import (
@@ -28,11 +35,9 @@ from huggingface_hub import (
28
 
29
  from diffusers.utils import load_image
30
 
31
- import spaces
32
  from huggingface_hub import HfApi
33
-
34
  token = os.getenv("HF_TOKEN")
35
- good_vae = AutoencoderKL.from_pretrained(base_model, subfolder="vae", torch_dtype=dtype, use_auth_token=token).to(device)
36
 
37
  #---if workspace = local or colab---
38
 
 
6
  import logging
7
  import numpy as np
8
  from typing import Any, Dict, List, Optional, Union
9
+ import gradio as gr
10
+ import logging
11
  import torch
12
  from PIL import Image
13
+ import spaces
14
+ from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL, AutoPipelineForImage2Image
15
+ from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
16
+ from diffusers.utils import load_image
17
+ from huggingface_hub import hf_hub_download, HfFileSystem, ModelCard, snapshot_download
18
+ import requests
19
+ import pandas as pd
20
 
21
 
22
  from diffusers import (
 
35
 
36
  from diffusers.utils import load_image
37
 
 
38
  from huggingface_hub import HfApi
 
39
  token = os.getenv("HF_TOKEN")
40
+
41
 
42
  #---if workspace = local or colab---
43