Eyalgut commited on
Commit
70ac0f8
·
verified ·
1 Parent(s): 36bd960

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import os
3
  hf_token = os.environ.get("HF_TOKEN")
4
  import spaces
5
  import torch
6
- from pipeline_bria import BriaPipeline
7
  import time
8
 
9
  resolutions = ["1024 1024","1280 768","1344 768","768 1344","768 1280"]
 
3
  hf_token = os.environ.get("HF_TOKEN")
4
  import spaces
5
  import torch
6
+ from pipeline_bria import BriaPipeline, BriaTransformer2DModel
7
  import time
8
 
9
  resolutions = ["1024 1024","1280 768","1344 768","768 1344","768 1280"]