Added in handling for blank label text

app.py

@@ -8,7 +8,15 @@ classifier = pipeline(model=checkpoint,
                       task="zero-shot-image-classification")
 
 def shot(image, labels_text):
-    labels = labels_text.split(";")
+    if len(labels_text):
+        labels = labels_text.split(";")
+    else:
+        labels = ['A page of printed text',
+                 'A page of handwritten text',
+                 'A blank page with no text',
+                 'A cover of a book',
+                 'A page of a book that contains a large illustration',
+                 'A page that features a table with multiple columns and rows']
     results = classifier(image,
                          candidate_labels=labels)
     return {result["label"]: result["score"] for result in results}