Fix generator script, add examples to widget sample card

README.md

@@ -4,6 +4,40 @@ language:
 tags:
 - font-identification
 license: mit
+widget:
+- text: "What font is this?"
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/ArchitectsDaughter-Regular_1.png
+  example_title: Architects Daughter
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Arial_Bold_39.png
+  example_title: Arial Bold
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Courier_28.png
+  example_title: Courier
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Helvetica_3.png
+  example_title: Helvetica
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/IBMPlexSans-Regular_25.png
+  example_title: IBM Plex Sans
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Inter-Regular_43.png
+  example_title: Inter
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Lobster-Regular_25.png
+  example_title: Lobster
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Merriweather-Regular_1.png
+  example_title: Merriweather
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Poppins-Regular_22.png
+  example_title: Poppins
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/RobotoMono-Regular_38.png
+  example_title: Roboto Mono
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Times_New_Roman_Bold Italic_26.png
+  example_title: Times New Roman Bold Italic
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Times_New_Roman_Italic_16.png
+  example_title: Times New Roman Italic
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/TitilliumWeb-Regular_5.png
+  example_title: Titillium Web
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Trebuchet_MS_Italic_47.png
+  example_title: Trebuchet MS Italic
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Trebuchet_MS_11.png
+  example_title: Trebuchet MS
+- src: https://huggingface.co/gaborcselle/font-identify/hf_sample/Verdana Bold_43.png
+  example_title: Verdana Bold
 ---
 
 # Font Identifier Project
@@ -20,7 +54,3 @@ Generate sample images (note this will work only on Mac): [gen_sample_data.py](g
 Arrange test images into test and train: [arrange_train_test_images.py](arrange_train_test_images.py)
 
 Train a ResNet18 on the data: [train_font_identifier.py](train_font_identifier.py)
-
-License: MIT
-
-

gen_sample_data.py

@@ -49,19 +49,20 @@ def main():
         # Output the font name so we can see the progress
         print(font_path, font_name)
 
-        # Random font size
-        font_size = random.choice(range(18, 72))
-
-        if font_path.endswith('.ttc'):
-            # ttc fonts have multiple fonts in one file, so we need to specify which one we want
-            font = ImageFont.truetype(font_path, font_size, index=0)
-        else:
-            # ttf fonts have only one font in the file
-            font = ImageFont.truetype(font_path, font_size)
 
         # Counter for the image filename
         j = 0
         for i in range(IMAGES_PER_FONT):  # Generate 50 images per font - reduced to 10 for now to make things faster
+            # Random font size
+            font_size = random.choice(range(18, 72))
+
+            if font_path.endswith('.ttc'):
+                # ttc fonts have multiple fonts in one file, so we need to specify which one we want
+                font = ImageFont.truetype(font_path, font_size, index=0)
+            else:
+                # ttf fonts have only one font in the file
+                font = ImageFont.truetype(font_path, font_size)
+
             # Determine the number of words that will fit on a line
             font_avg_char_width = font.getbbox('x')[2]
             words_per_line = int(800 / (font_avg_char_width*5))