Update app.py

app.py

@@ -9,8 +9,11 @@ model = RobertaForSequenceClassification.from_pretrained(model_path)
 
 # Define the prediction function
 def classify_text(text):
-    # Tokenize the input text
-    inputs = tokenizer(text, return_tensors='pt', padding=True, truncation=True, max_length=128)
+    # Remove periods and new lines from the input text
+    cleaned_text = text.replace('.', '').replace('\n', ' ')
+    
+    # Tokenize the cleaned input text
+    inputs = tokenizer(cleaned_text, return_tensors='pt', padding=True, truncation=True, max_length=128)
     
     # Get the model's prediction
     with torch.no_grad():