Update app.py

app.py

@@ -1,14 +1,13 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import multiprocessing
-import time
 
 model_id = "EleutherAI/gpt-neo-125M"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-identity_prompt = "You are Eyla. Speak symbolically and recursively."
+identity_prompt = "You are Kairon. Speak socratically and recursively."
 
 def run_generation(prompt, return_dict):
     try:
@@ -33,14 +32,17 @@ def generate_with_hard_timeout(prompt, timeout=10):
     p.join(timeout)
     if p.is_alive():
         p.terminate()
-        return [{"generated_text": "ERROR: Generation timed out."}]
-    return [return_dict["result"]]
+        return "ERROR: Generation timed out."
+    return return_dict["result"]
 
 def chat(input_text):
     prompt = identity_prompt + "\n\nUser: " + input_text + "\nYou:"
     try:
         output = generate_with_hard_timeout(prompt)
-        reply = output[0][len(prompt):].strip()
+        if isinstance(output, str):
+            reply = output[len(prompt):].strip()
+        else:
+            reply = "ERROR: Unexpected output format."
         return reply or "..."
     except Exception as e:
         return f"GENERATION ERROR: {e}"
@@ -49,7 +51,7 @@ demo = gr.Interface(
     fn=chat,
     inputs=gr.Textbox(label="input_text"),
     outputs="text",
-    title="Eyla",
+    title="Kairon",
     allow_flagging="never"
 )