update

app.py

@@ -125,10 +125,11 @@ def interface():
         with gr.Row():
             transformer_model_dropdown = gr.Dropdown(
                 label="Select Transformers Model",
-                choices=["TinyLlama/TinyLlama_v1.1"],  # Replace with actual models
+                choices=["TinyLlama/TinyLlama_v1.1", "HuggingFaceTB/SmolLM-360M"],  # Replace with actual models
                 value="TinyLlama/TinyLlama_v1.1",
                 interactive=True
             )
+            input_text_tr = gr.Texbox(label="Input Text", placeholder="Enter your input text here...")
             compare_button = gr.Button("Run Transformers Inference", elem_id="compare-button")
             transformer_result_output = gr.Textbox(label="Transformers Output", interactive=False, placeholder="Transformers output will appear here...")
             transformer_time_output = gr.Textbox(label="Transformers Inference Time", interactive=False, placeholder="Transformers inference time will appear here...")
@@ -136,7 +137,7 @@ def interface():
         # Actions
         setup_button.click(setup_bitnet, inputs=model_dropdown, outputs=setup_status)
         infer_button.click(run_inference, inputs=[model_dropdown, input_text, num_tokens], outputs=[result_output, time_output])
-        compare_button.click(run_transformers, inputs=[transformer_model_dropdown, input_text, num_tokens], outputs=[transformer_result_output, transformer_time_output])
+        compare_button.click(run_transformers, inputs=[transformer_model_dropdown, input_text_tr, num_tokens], outputs=[transformer_result_output, transformer_time_output])
 
 # Launch the Gradio app    
     return demo