hotbiz commited on
Commit
30a5017
1 Parent(s): 4b5fd6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,6 +1,10 @@
1
  import gradio as gr
2
 
3
- gr.load("models/black-forest-labs/FLUX.1-schnell").launch()
 
 
 
 
4
 
5
  dtype = torch.bfloat16
6
  device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -116,5 +120,4 @@ with gr.Blocks(css=css) as demo:
116
  outputs = [result, seed]
117
  )
118
 
119
-
120
  gr.load("models/black-forest-labs/FLUX.1-schnell").launch()
 
1
  import gradio as gr
2
 
3
+ import numpy as np
4
+ import random
5
+ import spaces
6
+ import torch
7
+ from diffusers import DiffusionPipeline
8
 
9
  dtype = torch.bfloat16
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
120
  outputs = [result, seed]
121
  )
122
 
 
123
  gr.load("models/black-forest-labs/FLUX.1-schnell").launch()