Update app.py

app.py

@@ -3,26 +3,16 @@ from huggingface_hub import InferenceClient
 
 client = InferenceClient("Intercept-Intelligence/Txt-Analyst-AI")
 
-def classify(text, max_tokens, temperature, top_p):
-    result = client.text_generation(
-        prompt=text,
-        max_tokens=max_tokens,
-        temperature=temperature,
-        top_p=top_p,
-    )
-    return result
+def classify(text):
+    result = client.text_classification(text)
+    return result[0]["label"]
 
 demo = gr.Interface(
     fn=classify,
-    inputs=[
-        gr.Textbox(label="Text to Analyze", placeholder="Enter text here..."),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p"),
-    ],
+    inputs=gr.Textbox(label="Text to Analyze", placeholder="Enter text here..."),
     outputs=gr.Textbox(label="Classification Result"),
     title="Txt Analyst AI",
-    description="Enter any text and let Intercept's AI classify or analyze it."
+    description="Enter any text and let Intercept's AI classify it based on its training."
 )
 
 if __name__ == "__main__":