Update app.py

app.py

@@ -20,6 +20,7 @@
 
 # Libraries
 import torch
+import pickle
 from utils import *
 import gradio as gr
 from numpy import array
@@ -37,9 +38,7 @@ nms_thresh = 0.30
 run_cuda = False
 
 # CFG Files
-cfg     = 'cfg/yolov3-openimages.cfg'
 clsnames= 'cfg/openimages.names'
-weights = 'cfg/yolov3-openimages.weights'
 
 # Load classes
 classes = load_classes(clsnames)
@@ -47,10 +46,8 @@ num_classes = len(classes)
 
 # Set up the neural network
 print('Load Network')
-model = Darknet(cfg)
-
-print('Load Weights')
-model.load_weights(weights)
+with open('models/darknet.pkl', 'rb') as f:
+    model = pickle.load(f)
 
 print('Successfully loaded Network')