Update main.py

main.py

@@ -18,7 +18,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = model.to(device)
 
 # Open a thing for the API
-@app.route("/api", methods=["POST"])
+@app.post("/api")
 def receive_data():
   data = request.get_json()
   
@@ -39,8 +39,8 @@ def receive_data():
   return jsonify(answer_data)
 
 # Incase a normal browser opens the page
-@app.route("/")
+@app.get("/")
 def not_api():
   return render_template("index.html")
 
-app.run(debug=False, port=7860)
+app.run(debug=False, port=7860, load_dotenv=True)