Update app.py

app.py

@@ -25,7 +25,7 @@ def hello_world():
     url = request.args.get('url')
     
     if url is None:
-        return jsonify(error="Url is required", url=None, class=None)
+        return jsonify(error="Url is required", url=None, label=None)
     
     image = Image.open(requests.get(url, stream=True).raw)
     
@@ -36,4 +36,4 @@ def hello_world():
     # model predicts one of the 1000 ImageNet classes
     predicted_class_idx = logits.argmax(-1).item()
 
-    return jsonify(url=url, class=model.config.id2label[predicted_class_idx])
+    return jsonify(url=url, label=model.config.id2label[predicted_class_idx])