Spaces:

Lod34
/

Animator2D-v1.0.0

Running

Lod34 commited on Apr 4

Commit

3e00e5f

verified ·

1 Parent(s): 2268f5b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,7 +55,7 @@ def generate_animation(description, base_frame_image, num_frames):
         model = Animator2DModel().to(device)
         logger.info("Modello inizializzato, caricamento pesi...")
         model.load_state_dict(torch.hub.load_state_dict_from_url(
-            "https://huggingface.co/Lod34/Animator2D-v1.0.0/resolve/main/animator2d_v1_0_0.pth",
             map_location=device))
         model.eval()
         logger.info("Modello caricato con successo")

         model = Animator2DModel().to(device)
         logger.info("Modello inizializzato, caricamento pesi...")
         model.load_state_dict(torch.hub.load_state_dict_from_url(
+            "https://huggingface.co/Lod34/Animator2D-v1.0.0/blob/main/Animator2D-v1.0.0.pth",
             map_location=device))
         model.eval()
         logger.info("Modello caricato con successo")