Update app.py

trust_remote_code=True

app.py

@@ -14,7 +14,7 @@ def get_model_summary(model_name):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     
     # Load the model and move it to the selected device
-    model = AutoModel.from_pretrained(model_name).to(device)
+    model = AutoModel.from_pretrained(model_name, trust_remote_code=True).to(device)
     
     # Return the model's architecture as a string
     return str(model)