Update app.py

app.py

@@ -14,26 +14,26 @@ import json
 import gradio as gr
 from recommender.main import get_tgi_config
 
-
-def greet(model_id, gpu_memory):
-    config = get_tgi_config(model_id, gpu_memory)
-    return json.dumps(config)
+def greet(model_id):
+    configs = get_recommendation(model_id)
+    return json.dumps(configs)
 
 
 demo = gr.Interface(
     fn=greet,
     inputs=[
         gr.Textbox(label="Model ID", placeholder="meta-llama/Llama-2-7b-chat-hf"),
-        gr.Slider(
-            step=4,
-            minimum=16,
-            maximum=640,
-            value=24,
-            label="GPU memory",
-            info="Select how much GPU memory you have available",
-        ),
+        # gr.Slider(
+        #     step=4000,
+        #     minimum=16_000,
+        #     maximum=640_000,
+        #     value=24_000,
+        #     label="GPU memory",
+        #     info="Select how much GPU memory you have available",
+        # ),
     ],
     outputs=[gr.JSON()],
 )
 
 demo.launch()
+