aifeifei798 commited on
Commit
134f180
·
verified ·
1 Parent(s): ac6eac5

Upload 4 files

Browse files
Files changed (1) hide show
  1. feifeilib/feifeiflorencebase.py +1 -1
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
- #image = Image.fromarray(image) # Convert NumPy array to PIL Image
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)