UI changes

app.py

@@ -142,7 +142,7 @@ with gr.Blocks(css=css) as demo:
 
         with gr.Column(scale=0.7, min_width=400):
             image_btn = gr.UploadButton("🖼️ Image Input", file_types=["image"])
-            text_input = gr.Textbox(label="Chat Input", lines=1, placeholder="Upload an image above [optional]. Then enter a text prompt, and press enter!")
+            text_input = gr.Textbox(label="Type a message", lines=1, placeholder="Upload an image above [optional]. Then enter a text prompt, and press enter!")
 
             with gr.Row():
                 with gr.Column(scale=0.33):

fromage/models.py

@@ -585,8 +585,7 @@ class Fromage(nn.Module):
             seen_image_idx.append(img_idx)
             img = utils.get_image_from_url(self.path_array[img_idx])
             image_outputs.append(img)
-            if len(image_outputs) == max_num_rets:
-              break
+            break
           except UnidentifiedImageError:
             pass
 

fromage/utils.py

@@ -32,6 +32,7 @@ def get_image_from_url(url: str):
 
 def truncate_caption(caption: str) -> str:
   """Truncate captions at periods and newlines."""
+  caption = caption.strip()
   trunc_index = caption.find('\n') + 1
   if trunc_index <= 0:
       trunc_index = caption.find('.') + 1