initial commit

app.py

@@ -0,0 +1,63 @@
+import gradio as gr
+import torch
+import random
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+# Load the trained model and tokenizer
+model_name = "your-username/hate-speech-classifier"  # Replace with your actual Hugging Face model repo
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model.to(device)
+model.eval()
+
+# Predefined usernames and chat messages for a game scenario
+usernames = ["ShadowSlayer", "DragonKnight", "PixelMage", "CyberRogue", "PhantomArcher"]
+game_responses = [
+    "I need backup at the base!",
+    "Watch out for enemies on the left!",
+    "Let's team up and attack together.",
+    "Great shot! That was amazing!",
+    "We need to capture the objective now!",
+    "Healing incoming, stay close!",
+    "I got eliminated, need a revive!",
+    "Nice strategy, let's keep it up!"
+]
+
+# Function for classification
+def classify_message(message):
+    inputs = tokenizer(message, padding="max_length", truncation=True, return_tensors="pt").to(device)
+    with torch.no_grad():
+        outputs = model(**inputs)
+        logits = outputs.logits
+        prediction = torch.argmax(logits, dim=1).item()
+    return "Hate speech/Offensive" if prediction == 1 else "Not hate speech/Offensive"
+
+# Chat simulation function
+def chat_interface(history, _):
+    username = random.choice(usernames)
+    new_message = random.choice(game_responses)
+    classification = classify_message(new_message)
+    blurred_message = "****" if classification == "Hate speech/Offensive" else new_message
+    history.append((f"{username}: {new_message}", f"{username}: {blurred_message}"))
+    
+    # Generate automated game response
+    bot_username = "GameMaster"
+    bot_response = random.choice(game_responses)
+    history.append((f"{bot_username}: {bot_response}", f"{bot_username}: {bot_response}"))
+    
+    return history, history
+
+# Create Gradio interface
+def main():
+    with gr.Blocks() as app:
+        gr.Markdown("# Game Chat Hate Speech Detection Simulator")
+        chatbot = gr.Chatbot()
+        submit = gr.Button("Generate Message")
+        
+        submit.click(chat_interface, inputs=[chatbot, None], outputs=[chatbot, chatbot])
+    
+    app.launch()
+
+if __name__ == "__main__":
+    main()