Update app.py

app.py

@@ -8,7 +8,7 @@ classifier = pipeline(model="Yanni8/star-predictor", token=token)
 
 
 def predict(text):
-    labels = classifier(text, top_k=5)[0]
+    labels = classifier(text, return_all_scores=True)[0]
     return  {label['label']: label['score'] for label in labels}
 
 iface = gr.Interface(