Update app.py

app.py

@@ -4,7 +4,7 @@ import time
 import random
 
 # Load the model and tokenizer
-model_id = "microsoft/phi-2"  # Change to your desired model
+model_id = "microsoft/phi-2"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)
 
@@ -31,12 +31,19 @@ def analyze_text(text):
         "Number of tokens": num_tokens,
     }
 
+def update_analysis(response):
+    analysis = analyze_text(response)
+    analysis_str = f"Number of characters: {analysis['Number of characters']}<br>" \
+                  f"Number of words: {analysis['Number of words']}<br>" \
+                  f"Number of tokens: {analysis['Number of tokens']}"
+    return analysis_str
+
 # --- Interface ---
 
 with gr.Blocks() as iface:
     gr.Markdown(
         """
-        # AI Chat
+        # Hajax Chat
         """
     )
 
@@ -48,7 +55,7 @@ with gr.Blocks() as iface:
     top_k = gr.Slider(label="Top k", minimum=1, maximum=100, step=1, value=50)
     max_length = gr.Slider(label="Max length", minimum=10, maximum=1000, step=10, value=250)
     submit_button = gr.Button(value="Submit")
-    response = gr.TextArea(label="AI Response:", lines=10)
+    response = gr.TextArea(label="Response:", lines=10)
     analysis_html = gr.HTML(elem_id="analysis")
 
     submit_button.click(fn=get_response, inputs=[input_text, temperature, top_p, top_k, max_length], outputs=[response])
@@ -67,15 +74,4 @@ with gr.Blocks() as iface:
     # Start a separate thread to update the background color
     gr.Interface.update(update_background, inputs=[], outputs=[], live=True)
 
-    # --- Analysis Logic ---
-
-    def update_analysis(response):
-        analysis = analyze_text(response)
-        analysis_str = f"Number of characters: {analysis['Number of characters']}<br>" \
-                      f"Number of words: {analysis['Number of words']}<br>" \
-                      f"Number of tokens: {analysis['Number of tokens']}"
-        iface.update(analysis=analysis_str, live=True)
-
-    iface.outputs[0].postprocess = update_analysis
-
 iface.launch(debug=True)