Update main.py

main.py

@@ -10,7 +10,7 @@ app = Flask(name)
 tokenizer = RobertaTokenizer.from_pretrained("microsoft/codebert-base")
 model = torch.load("model.pth", map_location=torch.device('cpu'))  # Load the trained model
 
-Ensure the model is in evaluation mode
+# Ensure the model is in evaluation mode
 model.eval()
 
 
@@ -53,6 +53,6 @@ def predict():
         return jsonify({"error": str(e)}), 500
 
 
-Run the Flask app
+# Run the Flask app
 if name == "main":
     app.run(debug=True)