Update app.py

app.py

@@ -3,7 +3,13 @@ import joblib
 import numpy as np
 import pandas as pd
 from openai import OpenAI
+from huggingface_hub import login
 from huggingface_hub import hf_hub_download
+import logfire
+
+login()
+
+logfire.configure(token=os.getenv("LOGFIRE_API_KEY"))
 
 # Load your pre-trained model and label names
 model_path = hf_hub_download(repo_id="govtech/zoo-entry-001", filename="model.joblib")
@@ -45,6 +51,7 @@ def classify_text(text):
         'Prediction': (probabilities[0] > 0.5).astype(int)
     })
     # Return an update to the DataFrame component to make it visible with the results
+    logfire.info(f"{text} ({probabilities[0]})")
     return gr.update(value=df, visible=True)
 
 with gr.Blocks(title="Zoo Entry 001") as iface: