Update app.py

app.py

@@ -12,6 +12,8 @@ model = AutoModelForSequenceClassification.from_pretrained("nlptown/bert-base-mu
 
 # Define prediction function
 def predict(texts):
+    if isinstance(texts, str):
+        texts = [texts]
     inputs = tokenizer(
         texts, 
         return_tensors="pt",