Update app.py
Browse files
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):
|