Spaces:
Running
on
Zero
Running
on
Zero
aifeifei798
commited on
Upload 4 files
Browse files
feifeilib/feifeiflorencebase.py
CHANGED
@@ -119,7 +119,7 @@ def process_image(image, task_prompt = "More Detailed Caption", text_input=None,
|
|
119 |
h_percent = (base_height / float(image.size[1]))
|
120 |
w_size = int((float(image.size[0]) * float(h_percent)))
|
121 |
image = image.resize((w_size, base_height), Image.LANCZOS)
|
122 |
-
|
123 |
if task_prompt == 'Caption':
|
124 |
task_prompt = '<CAPTION>'
|
125 |
results = run_example(task_prompt, image, model_id=model_id)
|
|
|
119 |
h_percent = (base_height / float(image.size[1]))
|
120 |
w_size = int((float(image.size[0]) * float(h_percent)))
|
121 |
image = image.resize((w_size, base_height), Image.LANCZOS)
|
122 |
+
|
123 |
if task_prompt == 'Caption':
|
124 |
task_prompt = '<CAPTION>'
|
125 |
results = run_example(task_prompt, image, model_id=model_id)
|