Added deployment updates.

app.py

@@ -173,14 +173,15 @@ def plagiarism_check(
 """
 AI DETECTION SECTION
 """
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 text_bc_model_path = "polygraf-ai/ai-text-bc-bert-2-7m"
 text_bc_tokenizer = AutoTokenizer.from_pretrained(text_bc_model_path)
-text_bc_model = AutoModelForSequenceClassification.from_pretrained(text_bc_model_path)
+text_bc_model = AutoModelForSequenceClassification.from_pretrained(text_bc_model_path).to(device)
 
 text_mc_model_path = "polygraf-ai/ai-text-mc-v5-lighter-spec"
 text_mc_tokenizer = AutoTokenizer.from_pretrained(text_mc_model_path)
-text_mc_model = AutoModelForSequenceClassification.from_pretrained(text_mc_model_path)
+text_mc_model = AutoModelForSequenceClassification.from_pretrained(text_mc_model_path).to(device)
 
 def remove_special_characters(text):
     cleaned_text = re.sub(r'[^a-zA-Z0-9\s]', '', text)
@@ -297,7 +298,6 @@ subprocess.run(command)
 nlp = spacy.load("en_core_web_sm")
 
 # for perplexity
-device = "cuda" if torch.cuda.is_available() else "cpu"
 model_id = "gpt2"
 gpt2_model = GPT2LMHeadModel.from_pretrained(model_id).to(device)
 gpt2_tokenizer = GPT2TokenizerFast.from_pretrained(model_id)
@@ -557,4 +557,4 @@ with gr.Blocks() as demo:
 
     date_from = ""
     date_to = ""
-demo.launch(share=True, server_name="0.0.0.0", server_port=80)
+demo.launch(share=True, server_name="0.0.0.0", server_port=7860)

update_and_restart.sh

@@ -0,0 +1,17 @@
+
+#!/bin/bash
+
+# Navigate to your app's directory
+cd /home/aliasgarov/copyright_checker
+
+# Pull the latest changes from the main branch
+git pull origin main
+
+# Kill the running Gradio 
+pkill -f "app.py"
+
+# Activate your Python environment if necessary
+# source /path/to/your/venv/bin/activate
+
+# Start your Gradio app (adjust the command as needed)
+nohup python app.py &

webhooks.json

@@ -0,0 +1,7 @@
+[
+  {
+    "id": "update-gradio-app",
+    "execute-command": "/home/aliasgarov/copyright_checker/update_and_restart.sh",
+    "command-working-directory": "/home/aliasgarov/copyright_checker/app.py"
+  }
+]