Update app.py

app.py

@@ -21,8 +21,7 @@ async def predict(data: InputData):
         a = input_data[0]
         b = input_data[1]
         c = sumar(a,b)
-        prediction = c
-        #return {"prediction": prediction.tolist()}
+        prediction = c        
         return {"prediction": prediction}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))