made API legal and hidden

bro jus tread the title of the commit

app.py

@@ -1,34 +1,47 @@
+import os
+import requests
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# Load a lightweight GPT model
-model_name = "EleutherAI/gpt-neo-125M"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-neo-125M"
+
+HEADERS = {
+    "Authorization": f"Bearer {os.environ['HUGGINGFACEHUB_API_TOKEN']}"
+}
 
 chat_history = []
 
 def chat(user_message):
-    chat_history.append(f"You: {user_message}")
-    prompt = "\n".join(chat_history) + "\nAI:"
-    inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(
-        **inputs,
-        max_length=inputs["input_ids"].shape[1] + 60,
-        pad_token_id=tokenizer.eos_token_id,
-        do_sample=True,
-        top_k=50,
-        top_p=0.95,
-        temperature=0.7,
-    )
-    full_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    reply = full_output[len(prompt):].strip()
-    chat_history.append(f"AI: {reply}")
+    chat_history.append({"role": "user", "content": user_message})
+    payload = {
+        "inputs": {
+            "past_user_inputs": [m["content"] for m in chat_history if m["role"] == "user"][:-1],
+            "generated_responses": [m["content"] for m in chat_history if m["role"] == "assistant"],
+            "text": user_message
+        },
+        "options": {"use_cache": False}
+    }
+
+    response = requests.post(API_URL, headers=HEADERS, json=payload)
+    response.raise_for_status()
+    result = response.json()
+
+    if isinstance(result, dict) and "generated_text" in result:
+        reply = result["generated_text"]
+    else:
+        reply = result[0].get("generated_text", "")
+
+    reply = reply.strip()
+    chat_history.append({"role": "assistant", "content": reply})
     return reply
 
 with gr.Blocks() as demo:
     chatbot_ui = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Type a message and press Enter")
-    msg.submit(chat, msg, chatbot_ui)
+    msg = gr.Textbox(
+        placeholder="Type your message and press Enter",
+        lines=1,
+        show_label=False
+    )
+    msg.submit(fn=chat, inputs=msg, outputs=chatbot_ui)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()