Update app.py
Browse files
app.py
CHANGED
@@ -56,7 +56,7 @@ def input_image_setup(file_loc):
|
|
56 |
]
|
57 |
return image_parts
|
58 |
|
59 |
-
def generate_gemini_response(input_prompt
|
60 |
image_prompt = input_image_setup(image_loc)
|
61 |
prompt_parts = [input_prompt image_prompt[0]]
|
62 |
response = model.generate_content(prompt_parts)
|
@@ -68,7 +68,7 @@ def upload_file(files, social_media_platform, num_captions=1):
|
|
68 |
if not files:
|
69 |
return None, "Image not uploaded"
|
70 |
file_paths = [file.name for file in files]
|
71 |
-
response = generate_gemini_response(input_prompt
|
72 |
return file_paths[0], response
|
73 |
|
74 |
with gr.Blocks() as demo:
|
|
|
56 |
]
|
57 |
return image_parts
|
58 |
|
59 |
+
def generate_gemini_response(input_prompt, image_loc):
|
60 |
image_prompt = input_image_setup(image_loc)
|
61 |
prompt_parts = [input_prompt image_prompt[0]]
|
62 |
response = model.generate_content(prompt_parts)
|
|
|
68 |
if not files:
|
69 |
return None, "Image not uploaded"
|
70 |
file_paths = [file.name for file in files]
|
71 |
+
response = generate_gemini_response(input_prompt, file_paths[0])
|
72 |
return file_paths[0], response
|
73 |
|
74 |
with gr.Blocks() as demo:
|