vs4vijay commited on
Commit
cf74689
·
1 Parent(s): 8bce86e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  import transformers
3
 
4
- # Load the Dall-E model
5
- dalle2 = transformers.DALL-E2.from_pretrained('dall-e')
6
 
7
  # Load the Stable Diffusion model
8
  stable_diffusion = transformers.StableDiffusion.from_pretrained('stable-diffusion')
@@ -24,12 +24,12 @@ def generate_text(input_text, model):
24
  return output_text
25
 
26
  # Create a Gradio interface for the Dall-E2 model
27
- dalle2_interface = gr.Interface(fn=generate_text,
28
- inputs=gr.inputs.Textbox(lines=1, label='Enter your text for Dall-E2'),
29
- outputs=gr.outputs.Textbox(label='Generated text'),
30
- description='Dall-E2 Text Generation',
31
- model=dalle2)
32
- dalle2_interface.launch()
33
 
34
  # Create a Gradio interface for the Stable Diffusion model
35
  stable_diffusion_interface = gr.Interface(fn=generate_text,
 
1
  import gradio as gr
2
  import transformers
3
 
4
+ # Load the Dall-E2 model
5
+ # dalle2 = transformers.DALL_E2.from_pretrained('dall-e')
6
 
7
  # Load the Stable Diffusion model
8
  stable_diffusion = transformers.StableDiffusion.from_pretrained('stable-diffusion')
 
24
  return output_text
25
 
26
  # Create a Gradio interface for the Dall-E2 model
27
+ # dalle2_interface = gr.Interface(fn=generate_text,
28
+ # inputs=gr.inputs.Textbox(lines=1, label='Enter your text for Dall-E2'),
29
+ # outputs=gr.outputs.Textbox(label='Generated text'),
30
+ # description='Dall-E2 Text Generation',
31
+ # model=dalle2)
32
+ # dalle2_interface.launch()
33
 
34
  # Create a Gradio interface for the Stable Diffusion model
35
  stable_diffusion_interface = gr.Interface(fn=generate_text,