Update app.py

app.py

@@ -5,10 +5,14 @@ api = HfApi()
 models=[
     "xai-org/grok-1"
 ]
-def load_models():
+def load_models(model_in):
     loaded_model=[]
     model_details=[]
-    similar_models = api.list_models(search=models[0].split("/")[1],limit=100,cardData=True)
+    
+    if "/" in models:
+        similar_models = api.list_models(search=model_in.split("/")[1],limit=100,cardData=True)
+    else:
+        similar_models = api.list_models(search=model_in,limit=100,cardData=True)
     for model in similar_models:
         try:
             loaded_model.append(gr.load(f'models/{model.id}'))
@@ -20,6 +24,8 @@ def load_models():
             model_details.append({"MODEL":model.id,"ERROR":ee})
     return loaded_model, model_details
 with gr.Blocks() as app:
+    with gr.Row():
+        model_name=gr.Textbox(label="Model", value=models[0], placeholder=models[0])
     with gr.Row():
         models_out=gr.JSON(label="Gradio Details")
         details=gr.JSON(label="Hub Details")