Update app.py

app.py

@@ -1,3 +1,16 @@
 import gradio as gr
+from transformers import pipeline
 
-gr.Interface.load("Yanni8/star-predictor").launch()
+classifier = pipeline(model="Yanni8/star-predictor")
+
+def predict(text):
+    labels = classifier(text, return_all_scores=True)[0]
+    return  {label['label']: label['score'] for label in labels}
+
+iface = gr.Interface(
+    fn=predict, 
+    inputs=gr.Textbox(lines=7, label="Input Text"), 
+    outputs=gr.Label(num_top_classes=3, label="Predicted Star")
+)
+
+iface.launch()