Swap to custom/trained model

app.py

@@ -15,8 +15,8 @@ categories = [
 logging.debug("Categories to classify: " + repr(categories))
 
 classifier = pipeline("zero-shot-classification",
-                      model="google-bert/bert-base-uncased")
-                      # model="Xylor/bert-case-uncased-single-sampled")
+                      # model="google-bert/bert-base-uncased")
+                      model="Xylor/bert-case-uncased-single-sampled")
                       # model="facebook/bart-large-mnli")
 
 # sequence_to_classify = "one day I will see the world"