imsaikat commited on
Commit
495bc09
·
verified ·
1 Parent(s): 9edf40f

Update app.py

Browse files

Added import torch message

Files changed (1) hide show
  1. app.py +8 -3
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=[run_button.click, prompt.submit],
31
  fn = infer,
32
- inputs = [prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
33
  outputs = [result, seed]
34
  )
35
 
36
- demo.launch()
 
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()