Add application file

app.py

@@ -46,7 +46,7 @@ def predict():
 
         predicted_class_idx = logits.argmax(-1).item()
         print(model.config.id2label[predicted_class_idx])
-        # Return the predictions
+        # Return the Predictions
         return json.dumps({"class": model.config.id2label[predicted_class_idx]})
     except Exception as e:
         return json.dumps({"Uh oh": f"{str(e)}"})