RanjithkumarPanjabikesan commited on
Commit
a3c6d7b
·
verified ·
1 Parent(s): 86bf6fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -7,6 +7,7 @@ from diffusers import DiffusionPipeline
7
  imagegenpipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
8
 
9
  #Define Image Captioner and Image Generate Functions
 
10
  def captioner(image):
11
  base64_image = image_to_base64_str(image)
12
  result = imagecaptioningpipe(base64_image)
@@ -15,8 +16,6 @@ def captioner(image):
15
  def generate(prompt):
16
  return imagegenpipe(prompt).images[0]
17
 
18
- from helper import image_to_base64_str, base64_to_pil
19
-
20
  #Build Image Caption and Generate Image Game App.
21
  import gradio as gr
22
  def caption_and_generate(image):
 
7
  imagegenpipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
8
 
9
  #Define Image Captioner and Image Generate Functions
10
+ from helper import image_to_base64_str, base64_to_pil
11
  def captioner(image):
12
  base64_image = image_to_base64_str(image)
13
  result = imagecaptioningpipe(base64_image)
 
16
  def generate(prompt):
17
  return imagegenpipe(prompt).images[0]
18
 
 
 
19
  #Build Image Caption and Generate Image Game App.
20
  import gradio as gr
21
  def caption_and_generate(image):