Update app.py

app.py

@@ -243,7 +243,7 @@ def inference(
                 loaded_learned_embeds = torch.load(file, map_location="cpu")
             else:
                 loaded_learned_embeds = load_file(file, device="cpu")
-            loaded_learned_embeds = loaded_learned_embeds["string_to_param"]["*"]
+            loaded_learned_embeds = loaded_learned_embeds["string_to_param"]["*"] if "string_to_param" in loaded_learned_embed else loaded_learned_embed
             ti_embs[name] = loaded_learned_embeds
 
         if len(ti_embs) > 0: