Update app.py

- revert original

app.py

@@ -66,26 +66,20 @@ model_input = gr.Textbox(
     visible=True
 )
 
-with gr.Blocks() as demo:
-    # Add the markdown link at the top
-
-    counter = <img src="https://hits.seeyoufarm.com/api/count/incr/badge.svg?url=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fxyizko%2FHF_Model_Test&count_bg=%2320221E&title_bg=%23555555&icon=&icon_color=%23E7E7E7&title=views&edge_flat=false"/>
-    gr.Markdown(counter)
-    
-    # Define the chat interface
-    chatbot = gr.ChatInterface(
-        fn=respond,
-        title="HF Text Generation Model Tester",
-        description="Add token + model name",
-        additional_inputs=[
-            gr.Textbox(label="Token"),
-            gr.Textbox(label="Model Name"),
-            gr.Textbox(value="You are helpful AI chatbot who responds like a medieval knight who is extremely polite, noble and funny.", label="System Message"),
-            gr.Slider(1, 2048, value=512, label="Max Tokens"),
-            gr.Slider(0.1, 4.0, value=0.7, label="Temperature"),
-            gr.Slider(0.1, 1.0, value=0.95, label="Top-p"),
-        ],
-    )
+# Chat interface
+demo = gr.ChatInterface(
+    fn=respond,
+    title="HF Text Generation Model Tester",
+    description="Add token + model name",
+    additional_inputs=[
+        token_input,
+        model_input,
+        gr.Textbox(value="You are helpful AI chatbot who reponds like a mideval knight who is extremely polite, noble and funny.", label="System Message"),
+        gr.Slider(1, 2048, value=512, label="Max Tokens"),
+        gr.Slider(0.1, 4.0, value=0.7, label="Temperature"),
+        gr.Slider(0.1, 1.0, value=0.95, label="Top-p"),
+    ],
+)
 
 if __name__ == "__main__":
     demo.launch()