calebaryee321 commited on
Commit
e1dcb6f
1 Parent(s): ef8a2de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,7 +33,7 @@ def img_Generation(text):
33
  pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
34
  pipe = pipe.to("cuda")
35
  image = pipe(text, num_inference_steps = 150).images[0]
36
- image.save("img_1.png")
37
 
38
  return image
39
 
@@ -48,4 +48,4 @@ def transcribe(audio):
48
  gr.Interface(
49
  fn=transcribe,
50
  inputs=gr.Audio(source="microphone", type="filepath"),
51
- outputs="image",description="A Speech to Image Generation App Using OpenAI's Whisper",title= "Whisper2IMG").launch(share="True")
 
33
  pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
34
  pipe = pipe.to("cuda")
35
  image = pipe(text, num_inference_steps = 150).images[0]
36
+ #image.save("img_1.png")
37
 
38
  return image
39
 
 
48
  gr.Interface(
49
  fn=transcribe,
50
  inputs=gr.Audio(source="microphone", type="filepath"),
51
+ outputs="image",description="A Speech to Image Generation App Using OpenAI's Whisper and Stable Diffusion V.2",title= "Whisper2IMG").launch()