Update app.py

app.py

@@ -3,32 +3,8 @@ 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(0, 5):
-        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 grid-cols-1 gap-5 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())
-for i in range(len(model_list)):
-    model_list[i] = model_list[i][1:]
-
-best_model_list = [
+model_list = [
     "runwayml/stable-diffusion-v1-5",
     "CompVis/stable-diffusion-v1-4",
     "prompthero/openjourney",
@@ -36,8 +12,6 @@ best_model_list = [
     "dreamlike-art/dreamlike-diffusion-1.0"
 ]
 
-model_list = best_model_list + model_list
-
 def tune_video_predict(
     pipe_id: str,
     prompt: str,