Update hubconf.py

hubconf.py

@@ -1,18 +1,18 @@
 import torch as tr
 
-from dummy import DummyNet
-from vanilla import VanillaNet
+from dummy import DummyNet as DummyNetClass
+from vanilla import VanillaNet as VanillaNetClass
 
 
 def DummyNet():
-    model = DummyNet()
+    model = DummyNetClass()
     model.load_state_dict(tr.load("dummy_weights.bin", map_location="cpu"))
     model.eval()
     return model
 
 
 def VanillaNet():
-    model = VanillaNet()
+    model = VanillaNetClass()
     model.load_state_dict(tr.load("vanilla_weights.bin", map_location="cpu"))
     model.eval()
     return model