update app.py file

app.py

@@ -6,11 +6,11 @@ checkpoint = "openai/clip-vit-large-patch14"
 classifier = pipeline(model=checkpoint, task="zero-shot-image-classification")
 def get_best_label(predictions):
     max_score = 0
-    label = ""
+    label = None
     for p in predictions:
         if p['score'] > max_score:
             max_score = p['score']
-            label = p['label']
+            label = str(p['label'])
     return label, max_score
 
 st.markdown('<h1 style="color:black;">Document Classifier</h1>', unsafe_allow_html=True)
@@ -30,6 +30,6 @@ if upload is not None:
   c2.header('Output')
   predictions = classifier(image, candidate_labels=["invoice, receipt", "bank statement, financial statement", "credit report"])
   label, score = get_best_label(predictions)
-  c2.subheader('Predicted class :' + str(label)+ " with score: " + str(score))
+  c2.subheader('Predicted class: ' + str(label) + " with score: " + str(score))
   c2.subheader('Probabilites:')
   c2.write(str(predictions))