Update app.py

app.py

@@ -4,7 +4,7 @@ import time
 import random
 
 # Load the model and tokenizer
-model_id = "microsoft/phi-2"
+model_id = "microsoft/phi-2"  # Change to your desired model
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)
 
@@ -33,114 +33,23 @@ def analyze_text(text):
 
 # --- Interface ---
 
-css = """
-.gradio-container {
-  background-color: #000000; /* Black background */
-  font-family: 'Roboto', sans-serif;
-}
-
-.gradio-interface {
-  background-color: rgba(0, 0, 0, 0.8); /* Dark translucent background */
-  border: 3px solid #00FF00; /* Green border */
-  padding: 20px;
-  box-shadow: 0 0 10px rgba(0, 255, 0, 0.5); /* Green shadow */
-}
-
-.gradio-button {
-  background-color: #00FF00; /* Green button */
-  color: white;
-  border: none;
-  padding: 10px 20px;
-  font-size: 16px;
-  cursor: pointer;
-}
-
-.gradio-button:hover {
-  background-color: #00CC00; /* Darker green on hover */
-}
-
-.gradio-textbox {
-  background-color: #111111; /* Dark gray background */
-  color: #00FF00; /* Green text */
-  border: 1px solid #00FF00; /* Green border */
-  padding: 10px;
-  margin-bottom: 10px;
-}
-
-.gradio-text-area {
-  background-color: #111111; /* Dark gray background */
-  color: #00FF00; /* Green text */
-  border: 1px solid #00FF00; /* Green border */
-  padding: 10px;
-}
-
-.gradio-slider {
-  background-color: #111111; /* Dark gray background */
-  color: #00FF00; /* Green text */
-}
-
-.gradio-slider .slider-bar {
-  background-color: #00FF00; /* Green slider bar */
-}
-
-.gradio-slider .slider-thumb {
-  background-color: #00FF00; /* Green slider thumb */
-}
-
-.gradio-label {
-  color: #00FF00; /* Green labels */
-}
-
-h1, h2 {
-  color: #00FF00; /* Green headings */
-  text-align: center;
-}
-
-.analysis-container {
-  margin-top: 20px;
-  padding: 10px;
-  border: 1px solid #00FF00;
-}
-"""
-
 with gr.Blocks() as iface:
     gr.Markdown(
         """
-        # NVIDIA AI Chat
-        Engage in a conversation with our advanced AI model. Customize the response using various parameters.
+        # AI Chat
         """
     )
 
-    with gr.Row():
-        input_text = gr.Textbox(
-            label="Your message:", lines=5, placeholder="Ask me anything...", show_label=True
-        )
-        gr.HTML(
-            f"""
-            <style>
-                .gradio-textbox {
-                    background-color: #111111;
-                    color: #00FF00;
-                    border: 1px solid #00FF00;
-                }
-            </style>
-            """, elem_id="textbox_styles"
-        )
-
-    with gr.Row():
-        temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, step=0.1, value=0.7)
-        top_p = gr.Slider(label="Top p", minimum=0.1, maximum=1.0, step=0.1, value=0.9)
-
-    with gr.Row():
-        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)
-
-    with gr.Row():
-        submit_button = gr.Button(value="Submit")
-
-    with gr.Row():
-        response = gr.TextArea(label="AI Response:", lines=10)
-        analysis_html = gr.HTML(elem_id="analysis")
+    input_text = gr.Textbox(
+        label="Your message:", lines=5, placeholder="Ask me anything...", show_label=True
+    )
+    temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, step=0.1, value=0.7)
+    top_p = gr.Slider(label="Top p", minimum=0.1, maximum=1.0, step=0.1, value=0.9)
+    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)
+    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])
     response.change(fn=update_analysis, inputs=[response], outputs=[analysis_html])
@@ -162,9 +71,9 @@ with gr.Blocks() as iface:
 
     def update_analysis(response):
         analysis = analyze_text(response)
-        analysis_str = f"<div class='analysis-container'>Number of characters: {analysis['Number of characters']}<br>" \
+        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']}</div>"
+                      f"Number of tokens: {analysis['Number of tokens']}"
         iface.update(analysis=analysis_str, live=True)
 
     iface.outputs[0].postprocess = update_analysis