Initial commit

README.md

@@ -4,7 +4,7 @@ emoji: 🐨
 colorFrom: purple
 colorTo: yellow
 sdk: gradio
-sdk_version: 4.43.0
+sdk_version: 3.50.0
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -0,0 +1,53 @@
+import gradio as gr
+import onnxruntime as ort
+from transformers import AutoTokenizer
+import numpy as np
+
+# Load the ONNX model
+onnx_model_path = "fine-tuned_all-distilroberta-v1_quantized.onnx"
+ort_session = ort.InferenceSession(onnx_model_path)
+
+# Load the tokenizer
+tokenizer = AutoTokenizer.from_pretrained(
+    "sentence-transformers/all-MiniLM-L6-v2")
+
+
+def predict_similarity(question, candidate_answer, ai_answer):
+    # Combine question and answers
+    candidate_combined = f"Question: {question} Answer: {candidate_answer}"
+    ai_combined = f"Question: {question} Answer: {ai_answer}"
+
+    # Tokenize inputs
+    inputs = tokenizer([candidate_combined, ai_combined],
+                       padding=True, truncation=True, return_tensors="np")
+
+    # Run inference
+    ort_inputs = {
+        "input_ids": inputs["input_ids"],
+        "attention_mask": inputs["attention_mask"]
+    }
+    ort_outputs = ort_session.run(None, ort_inputs)
+
+    # Calculate cosine similarity
+    embeddings = ort_outputs[0]
+    similarity = np.dot(embeddings[0], embeddings[1]) / \
+        (np.linalg.norm(embeddings[0]) * np.linalg.norm(embeddings[1]))
+
+    return float(similarity)
+
+
+# Create Gradio interface
+iface = gr.Interface(
+    fn=predict_similarity,
+    inputs=[
+        gr.Textbox(label="Coding Question"),
+        gr.Textbox(label="Candidate's Answer"),
+        gr.Textbox(label="AI-generated Answer")
+    ],
+    outputs=gr.Number(label="Similarity Score"),
+    title="AI Code Detector",
+    description="Detect similarity between human-written and AI-generated coding answers."
+)
+
+# Launch the app
+iface.launch()

fine-tuned_all-distilroberta-v1_quantized.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b3df341367812546bc086c5d3fa809bbde3a35dc89cc34cd5bff32dd0bd711f3
+size 82520985

requirements.txt

@@ -0,0 +1,4 @@
+gradio==3.50.0
+onnxruntime==1.19.2
+torch==2.4.1
+transformers==4.44.2