Deploy project

app.py

@@ -1,4 +1,4 @@
-ifrom flask import Flask, request, jsonify
+from flask import Flask, request, jsonify
 from flask_cors import CORS
 import torch
 from transformers import RobertaTokenizer, RobertaForSequenceClassification
@@ -6,12 +6,13 @@ from transformers import RobertaTokenizer, RobertaForSequenceClassification
 app = Flask(__name__)
 CORS(app)
 
+# Load model and tokenizer
 tokenizer = RobertaTokenizer.from_pretrained("microsoft/codebert-base")
 model = RobertaForSequenceClassification.from_pretrained("microsoft/codebert-base")
 
 @app.route("/")
 def home():
-    return "✅ Bug Detection and Fixing API is Running!"
+    return "Bug Detection and Fixing API is running!"
 
 @app.route("/detect", methods=["POST"])
 def detect_bug():
@@ -24,8 +25,9 @@ def detect_bug():
         inputs = tokenizer(code, return_tensors="pt", truncation=True, padding=True)
         outputs = model(input_ids=inputs["input_ids"], attention_mask=inputs["attention_mask"])
         prediction = torch.argmax(outputs.logits, dim=1).item()
-        status = "🐛 Buggy" if prediction == 1 else "✅ Clean"
-        return jsonify({"status": status})
+        bug_status = "buggy" if prediction == 1 else "clean"
+
+        return jsonify({"status": bug_status})
 
     except Exception as e:
         return jsonify({"error": str(e)}), 500

requirements.txt

@@ -1,5 +1,5 @@
-transformers
-torch
+transformers==4.52.4
+torch==2.7.1
+flask==3.1.1
+flask-cors==4.0.0
 gradio==4.44.1
-flask
-flask-cors