Update dummy_model.py

dummy_model.py

@@ -20,8 +20,21 @@ def create_dummy_model():
     model.fc = nn.Linear(model.fc.in_features, num_classes)
     
     # Save the untrained model
-    torch.save(model.state_dict(), 'plant_disease_model.pth')
-    print("Dummy model saved as plant_disease_model.pth")
+    try:
+        torch.save(model.state_dict(), 'plant_disease_model.pth')
+        print("Dummy model saved as plant_disease_model.pth")
+        
+        # Verify the file was created
+        if os.path.exists('plant_disease_model.pth'):
+            print(f"Verified: model file exists with size {os.path.getsize('plant_disease_model.pth')} bytes")
+        else:
+            print("Warning: Failed to create model file")
+    except Exception as e:
+        print(f"Error saving dummy model: {e}")
+        # Create an empty file as fallback
+        with open('plant_disease_model.pth', 'w') as f:
+            f.write('')
+        print("Created empty model file as fallback")
     
     # Create default class names
     class_names = [