Upload gradio_interface.py

gradio_interface.py

@@ -0,0 +1,43 @@
+# Import necessary libraries
+import gradio as gr
+import spacy
+
+# Load the trained spaCy model
+model_path = "./my_trained_model"
+nlp = spacy.load(model_path)
+
+# Function to classify text
+def classify_text(text):
+    doc = nlp(text)
+    predicted_labels = doc.cats
+    return predicted_labels
+
+# Function to save results to a file
+def save_to_file(text, predicted_labels):
+    with open("classification_results.txt", "w") as f:
+        f.write("Text: {}\n\n".format(text))
+        for label, score in predicted_labels.items():
+            f.write("{}: {}\n".format(label, score))
+
+# Gradio Interface
+inputs = [
+    gr.inputs.Textbox(lines=7, label="Enter your text"),
+    gr.inputs.File(label="Upload a file")
+]
+
+output = gr.outputs.Textbox(label="Classification Results")
+
+def classify_and_save(input_text, input_file):
+    if input_text:
+        text = input_text
+    elif input_file:
+        # Process the file and extract text
+        with open(input_file.name, "r") as f:
+            text = f.read()
+    
+    predicted_labels = classify_text(text)
+    save_to_file(text, predicted_labels)
+    return predicted_labels
+
+iface = gr.Interface(fn=classify_and_save, inputs=inputs, outputs=output, title="Text Classifier")
+iface.launch(share=True)