jynzo94 commited on
Commit
0688f83
·
1 Parent(s): 89fef56

update to use gradio

Browse files
Files changed (2) hide show
  1. .gradio/certificate.pem +31 -0
  2. app.py +19 -31
.gradio/certificate.pem ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ -----BEGIN CERTIFICATE-----
2
+ MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw
3
+ TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh
4
+ cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4
5
+ WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu
6
+ ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY
7
+ MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc
8
+ h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+
9
+ 0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U
10
+ A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW
11
+ T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH
12
+ B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC
13
+ B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv
14
+ KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn
15
+ OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn
16
+ jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw
17
+ qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI
18
+ rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV
19
+ HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq
20
+ hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL
21
+ ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ
22
+ 3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK
23
+ NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5
24
+ ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur
25
+ TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC
26
+ jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc
27
+ oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq
28
+ 4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA
29
+ mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d
30
+ emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc=
31
+ -----END CERTIFICATE-----
app.py CHANGED
@@ -1,44 +1,32 @@
1
- from flask import Flask, request, jsonify
2
  from transformers import pipeline
3
 
4
- # Initialize Flask app
5
- app = Flask(__name__)
6
-
7
  # Load language detection model from Hugging Face
8
  lang_classifier = pipeline("text-classification", model="papluca/xlm-roberta-base-language-detection")
9
 
10
  # Load hate speech detection model from Hugging Face
11
  hate_classifier = pipeline("text-classification", model="Hate-speech-CNERG/dehatebert-mono-english")
12
 
13
- # Route: Health check
14
- @app.route("/", methods=["GET"])
15
- def home():
16
- return jsonify({"message": "Language & Hate Speech API is running!"})
17
-
18
- # Route: Detect Language
19
- @app.route("/detect-language", methods=["POST"])
20
- def detect_language():
21
- data = request.json
22
- text = data.get("text", "").strip()
23
-
24
- if not text:
25
- return jsonify({"error": "No text provided"}), 400
26
 
27
- result = lang_classifier(text)
28
- return jsonify({"language": result[0]['label'], "confidence": result[0]['score']})
29
 
30
- # Route: Detect Hate Speech
31
- @app.route("/detect-hate-speech", methods=["POST"])
32
- def detect_hate_speech():
33
- data = request.json
34
- text = data.get("text", "").strip()
35
 
36
- if not text:
37
- return jsonify({"error": "No text provided"}), 400
38
 
39
- result = hate_classifier(text)
40
- return jsonify({"label": result[0]['label'], "score": result[0]['score']})
 
 
 
 
 
 
41
 
42
- # Run the server
43
- if __name__ == "__main__":
44
- app.run(host="0.0.0.0", port=7860)
 
1
+ import gradio as gr
2
  from transformers import pipeline
3
 
 
 
 
4
  # Load language detection model from Hugging Face
5
  lang_classifier = pipeline("text-classification", model="papluca/xlm-roberta-base-language-detection")
6
 
7
  # Load hate speech detection model from Hugging Face
8
  hate_classifier = pipeline("text-classification", model="Hate-speech-CNERG/dehatebert-mono-english")
9
 
10
+ def analyze_text(text):
11
+ if not text.strip():
12
+ return {"error": "No text provided"}, {"error": "No text provided"}
 
 
 
 
 
 
 
 
 
 
13
 
14
+ lang_result = lang_classifier(text)
15
+ hate_result = hate_classifier(text)
16
 
17
+ language_output = {"language": lang_result[0]['label'], "confidence": lang_result[0]['score']}
18
+ hate_output = {"label": hate_result[0]['label'], "score": hate_result[0]['score']}
 
 
 
19
 
20
+ return language_output, hate_output
 
21
 
22
+ # Define the Gradio interface
23
+ iface = gr.Interface(
24
+ fn=analyze_text, # Use the combined function
25
+ inputs=gr.Textbox(label="Enter text"),
26
+ outputs=[gr.JSON(label="Language Detection"), gr.JSON(label="Hate Speech Detection")],
27
+ title="Language & Hate Speech Detection",
28
+ description="Enter text to detect its language and check for hate speech."
29
+ )
30
 
31
+ # Launch the Gradio app
32
+ iface.launch(server_name="0.0.0.0", server_port=7860, share=True)