update app.py

app.py

@@ -416,35 +416,38 @@ def create_title_image(word1, word2, word3, width, font_path='./assets/OpenSans-
     # Create a new image with the specified width and white background
     image = Image.new('RGB', (width, 60), (255, 255, 255))
 
-    # Load the specified font
-    font = ImageFont.truetype(font_path, font_size)
-
-    # Keep increasing the font size until all words fit within the desired width
-    while True:
-        # Create a draw object for the image
-        draw = ImageDraw.Draw(image)
-        
-        word_spacing = font_size / 2
-        # Draw each word in the appropriate color
-        x_offset = word_spacing
-        draw.text((x_offset, 0), word1, color_red, font=font)
-        x_offset += font.getsize(word1)[0] + word_spacing
-        draw.text((x_offset, 0), word2, color_black, font=font)
-        x_offset += font.getsize(word2)[0] + word_spacing
-        draw.text((x_offset, 0), word3, color_blue, font=font)
-        
-        word_sizes = [font.getsize(word) for word in [word1, word2, word3]]
-        total_width = sum([size[0] for size in word_sizes]) + word_spacing * 3
-
-        # Stop increasing font size if the image is within the desired width
-        if total_width <= width:
-            break
-            
-        # Increase font size and reset the draw object
-        font_size -= 1
-        image = Image.new('RGB', (width, 50), (255, 255, 255))
+    try:
+        # Load the specified font
         font = ImageFont.truetype(font_path, font_size)
-        draw = None
+
+        # Keep increasing the font size until all words fit within the desired width
+        while True:
+            # Create a draw object for the image
+            draw = ImageDraw.Draw(image)
+            
+            word_spacing = font_size / 2
+            # Draw each word in the appropriate color
+            x_offset = word_spacing
+            draw.text((x_offset, 0), word1, color_red, font=font)
+            x_offset += font.getsize(word1)[0] + word_spacing
+            draw.text((x_offset, 0), word2, color_black, font=font)
+            x_offset += font.getsize(word2)[0] + word_spacing
+            draw.text((x_offset, 0), word3, color_blue, font=font)
+            
+            word_sizes = [font.getsize(word) for word in [word1, word2, word3]]
+            total_width = sum([size[0] for size in word_sizes]) + word_spacing * 3
+
+            # Stop increasing font size if the image is within the desired width
+            if total_width <= width:
+                break
+                
+            # Increase font size and reset the draw object
+            font_size -= 1
+            image = Image.new('RGB', (width, 50), (255, 255, 255))
+            font = ImageFont.truetype(font_path, font_size)
+            draw = None
+    except Exception as e:
+        pass
 
     return image
 
@@ -502,7 +505,7 @@ def relate_anything(input_image, k):
 
         current_pil_image = pil_image.copy()
         current_pil_image.alpha_composite(mask_image)
-        
+                
         title_image = create_title_image('Red', relation, 'Blue', current_pil_image.size[0])
         concate_pil_image = concatenate_images_vertical(current_pil_image, title_image)
         pil_image_list.append(concate_pil_image)
@@ -704,13 +707,13 @@ if __name__ == "__main__":
     args = parser.parse_args()
     print(f'args = {args}')
 
-    # set_device()
+    set_device()
     get_sam_vit_h_4b8939()
     load_groundingdino_model()
     load_sam_model()
-    # load_sd_model()
+    load_sd_model()
     load_lama_cleaner_model()
-    # load_ram_model()
+    load_ram_model()
 
     os.system("pip list")