Update app.py

app.py

@@ -3,7 +3,28 @@ from tuneavideo.models.unet import UNet3DConditionModel
 from tuneavideo.util import save_videos_grid
 import torch
 import gradio as gr
+from bs4 import BeautifulSoup
+import requests
 
+def model_url_list():
+    url_list = []
+    for i in range(1, 9):
+        url_list.append(f"https://huggingface.co/models?p={i}&sort=downloads&search=dreambooth")
+    return url_list
+
+
+def data_scraping(url_list):
+    model_list = []
+    for url in url_list:
+        response = requests.get(url)
+        soup = BeautifulSoup(response.text, "html.parser")
+        div_class = 'grid gap-5 grid-cols-1 2xl:grid-cols-2'
+        div = soup.find('div', {'class': div_class})
+        for a in div.find_all('a', href=True):
+            model_list.append(a['href'])
+    return model_list
+
+model_list = data_scraping(model_url_list())
 
 def tune_video_predict(
     pipe_id: str,
@@ -24,10 +45,7 @@ def tune_video_predict(
 demo_inputs = [
     gr.inputs.Dropdown(
         label="Model",
-        choices=[
-            "Tune-A-Video-library/a-man-is-surfing",
-            "sd-dreambooth-library/mr-potato-head",
-        ]
+        choices=model_list
     ),
     gr.inputs.Textbox(
         label="Prompt",