Create app.py

app.py

@@ -0,0 +1,96 @@
+import panel as pn
+from transformers_agent_ui import TransformersAgentUI
+from huggingface_hub import HfApi
+
+# Load the model list from the script
+model_list = [
+    "LayoutLMv3",
+    "LayoutXLM",
+    "LED",
+    "LeViT",
+    "LiLT",
+    "LLaMA",
+    "Llama2",
+    "Llama3",
+    "LLaVa",
+    "LLaVA-NeXT",
+    "LLaVA-NeXT-Video",
+    "LLaVA-Onevision",
+    "Longformer",
+    "LongT5",
+    "LUKE",
+    "LXMERT",
+    "M-CTC-T",
+    "M2M100",
+    "MADLAD-400",
+    "Mamba",
+    "mamba2",
+    "Marian",
+    "MarkupLM",
+    "Mask2Former",
+    "MaskFormer",
+    "MatCha",
+    "mBART",
+    "mBART-50",
+    "MEGA",
+    "Megatron-BERT",
+    "Megatron-GPT2",
+    "MGP-STR",
+    "Mimi",
+    "Mistral",
+    "Mixtral",
+    "Mllama",
+    "mLUKE",
+    "MMS",
+    "MobileBERT",
+    "MobileNetV1",
+    "MobileNetV2",
+    "MobileViT",
+    "MobileViTV2",
+    "Moshi",
+    "MPNet",
+    "MPT",
+    "MRA",
+    "MT5",
+    "MusicGen",
+]
+
+# Create a dropdown menu for model selection
+model_selector = pn.widgets.Select(options=model_list, value=model_list[0])
+
+# Create a text input for the repository list file
+repo_list_file = pn.widgets.TextInput(value="repo_list.txt")
+
+# Create a text input for the output directory
+output_dir = pn.widgets.TextInput(value="output_dir")
+
+# Create a checkbox for dry run
+dry_run = pn.widgets.Checkbox(value=False)
+
+# Create a button to start the process
+start_button = pn.widgets.Button(name="Start")
+
+# Define the function to be executed when the button is clicked
+def start_process(event):
+    # Get the selected model and repository list file
+    model = model_selector.value
+    repo_list_file_path = repo_list_file.value
+    output_dir_path = output_dir.value
+    dry_run_value = dry_run.value
+
+    # Call the script with the selected options
+    process_repos(output_dir_path, model, repo_list_file_path, dry_run_value)
+
+# Link the button to the function
+start_button.on_click(start_process)
+
+# Create the UI
+ui = pn.Column(
+    pn.Row(model_selector, repo_list_file, output_dir, dry_run),
+    start_button,
+)
+
+# Serve the UI
+if pn.state.served:
+    pn.extension("terminal", "notifications", notifications=True, design="bootstrap")
+    ui.servable()