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))