Update main.py

main.py

@@ -25,6 +25,6 @@ async def get_prediction():
     results = model('test.jpg') 
     # View results
     for r in results:
-        print(names[int(r.probs[top1])]) 
+        print(names[int(r.probs['top1'])]) 
 
     return r.probs