Update main.py

main.py

@@ -18,9 +18,8 @@ app.add_middleware(
 
 
 
-    
+names = {0: 'breaking-stage', 1: 'half-riping-stage', 2: 'overripe un-healthy', 3: 'ripe', 4: 'ripe_with_consumable_disease', 5: 'unripe'}   
 @app.get("/predict")
-names = {0: 'breaking-stage', 1: 'half-riping-stage', 2: 'overripe un-healthy', 3: 'ripe', 4: 'ripe_with_consumable_disease', 5: 'unripe'}
 async def get_prediction():
     # Predict with the model
     results = model('test.jpg') 
@@ -28,4 +27,4 @@ async def get_prediction():
     for r in results:
         print(names[int(r.probs[top1])]) 
 
-    return str(results)
+    return r.probs