Spaces:
Sleeping
Sleeping
Update app.py
Browse filesAdded import torch message
app.py
CHANGED
|
@@ -1,4 +1,9 @@
|
|
| 1 |
import gradio as gr
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
from transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
|
| 3 |
|
| 4 |
# gr.load("models/black-forest-labs/FLUX.1-dev").launch()
|
|
@@ -27,10 +32,10 @@ def infer(prompt, seed=42, randomize_seed=True, width=400, height=400, guidance_
|
|
| 27 |
return image, seed
|
| 28 |
|
| 29 |
gr.on(
|
| 30 |
-
triggers=
|
| 31 |
fn = infer,
|
| 32 |
-
inputs = [prompt
|
| 33 |
outputs = [result, seed]
|
| 34 |
)
|
| 35 |
|
| 36 |
-
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
+
import torch as torch
|
| 3 |
+
import numpy as np
|
| 4 |
+
|
| 5 |
+
|
| 6 |
+
from diffusers import DiffusionPipeline
|
| 7 |
from transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
|
| 8 |
|
| 9 |
# gr.load("models/black-forest-labs/FLUX.1-dev").launch()
|
|
|
|
| 32 |
return image, seed
|
| 33 |
|
| 34 |
gr.on(
|
| 35 |
+
triggers=None,
|
| 36 |
fn = infer,
|
| 37 |
+
inputs = [prompt],
|
| 38 |
outputs = [result, seed]
|
| 39 |
)
|
| 40 |
|
| 41 |
+
app.launch()
|