Update model to v16, fix name bug, add alternate render option for local.

app.py

@@ -45,7 +45,7 @@ def gen_card_text(name):
     if name == '':
       prompt = f"Name: {random.choice('ABCDEFGHIJKLMNOPQRSTUVWXYZ')}"
     else:
-      prompt = f"Name: {name}\r\n"
+      prompt = f"Name: {name}\n"
     print(f'GENERATING CARD TEXT with prompt: {prompt}')
     output = text_pipeline(prompt, max_length=512, num_return_sequences=1, num_beams=5, temperature=1.5, do_sample=True,
                            repetition_penalty=1.2, eos_token_id=eos_id)
@@ -211,10 +211,18 @@ def html_to_png(card_name, html):
 
     path = os.path.join('rendered_cards', save_name)
     try:
-      css = ['./colab-data-test/css/mana.css', './colab-data-test/css/keyrune.css', './colab-data-test/css/mtg_custom.css']
-      imgkit.from_string(html, path, {"xvfb": ""}, css=css)
+        css = ['./colab-data-test/css/mana.css', './colab-data-test/css/keyrune.css', './colab-data-test/css/mtg_custom.css']
+        imgkit.from_string(html, path, {"xvfb": ""}, css=css)
     except:
-      pass
+        # For Windows local, requires 'html2image' package from pip.
+        from html2image import Html2Image
+        rendered_card_dir = 'rendered_cards'
+        hti = Html2Image(output_path=rendered_card_dir)
+        paths = hti.screenshot(html_str=html,
+                               css_file=['./colab-data-test/css/mtg_custom.css', './colab-data-test/css/mana.css', './colab-data-test/css/keyrune.css'],
+                               save_as=save_name, size=(450, 600))
+        print(paths)
+        path = paths[0]
     print('OPENING IMAGE FROM FILE')
     img = Image.open(path)
     print('CROPPING BACKGROUND')

gpt2-magic-card/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c3b0f3c17995a67032a94e998a790f5ea1ae1dbfd9d231c8bc7625b6556cb7d1
+oid sha256:dee3589ff13b3b51415f3eec89cb6119f36688ae26a65c1f07d695297a919656
 size 248927581