changed load weights_only to false

app.py

@@ -36,7 +36,7 @@ model = None
 # Load model
 try:
     model = timm.create_model('vit_base_patch16_224', pretrained=False, num_classes=47)
-    model.load_state_dict(torch.load(model_path, weights_only=True))
+    model.load_state_dict(torch.load(model_path, weights_only=False))
     model.eval()
 except Exception as e:
     print(f"Error loading model: {str(e)}")
@@ -96,4 +96,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()