Update app.py

app.py

@@ -1,23 +1,17 @@
-import torch
-from transformers import pipeline
-import gradio as gr
 import pandas as pd
-from datasets import Dataset
+import gradio as gr
+from transformers import pipeline
+
+# URL of the CSV file
+csv_url = "https://huggingface.co/spaces/kataniccc/tekst/blob/main/submission.csv"
 
-# Enable SafeTensors if available
-if torch.__version__ >= "1.10":
-    torch.set_safety_enabled(True)
+# Read the CSV file into a pandas DataFrame
+df = pd.read_csv(csv_url)
 
-# Load the model
+# Load the model for text classification
 model_nm = 'microsoft/deberta-v3-small'
 classifier = pipeline("text-classification", model=model_nm)
 
-# Read and preprocess data
-df = pd.read_csv("path/to/train.csv")  # Replace "path/to/train.csv" with the actual path
-df.describe(include='object')
-df['input'] = 'TEXT1: ' + df.context + '; TEXT2: ' + df.target + '; ANC1: ' + df.anchor
-ds = Dataset.from_pandas(df)
-
 # Define prediction function
 def predict_text(input_text):
     prediction = classifier(input_text)