Update Endpoint

app.py

@@ -16,7 +16,7 @@ model = BertForSequenceClassification.from_pretrained("sleiyer/restricted_item_d
 tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
 
 # Function to predict the class of a single input text
-def predict(text):
+def predict(text: str):
   # Preprocess the input text
   inputs = tokenizer(text, return_tensors='pt', truncation=True, padding=True)