Update scripts/mainrunpodA1111.py

scripts/mainrunpodA1111.py

@@ -252,7 +252,7 @@ def modeldwn(model_LINK):
               print('Model already exists')
 
         if os.path.exists(loramodel) :
-          print('LoRA downloaded')
+          print('Checkpoints downloaded')
         else:
           print('Wrong link, check that the link is valid')
 
@@ -498,30 +498,9 @@ def getsrc(url):
     return src
 
 def get_true_name(url):
-    import requests
-    from bs4 import BeautifulSoup
-
-    # Отправляем запрос на страницу предпросмотра файла
-    response = requests.get(url)
-
-    # Проверяем, успешно ли выполнен запрос
-    if response.status_code != 200:
-        print(f"Failed to retrieve the page: {response.status_code}")
-        return None
-
-    # Используем BeautifulSoup для парсинга HTML
-    soup = BeautifulSoup(response.text, 'html.parser')
-
-    # Ищем тег <title>, который может содержать имя файла
-    title_tag = soup.find('title')
-    if title_tag:
-        # Имя файла может быть в формате "file_name.extension - Google Drive" в теге <title>
-        # Поэтому мы разделяем строку по ' - ' и берем первую часть
-        file_name = title_tag.text.split(' - ')[0]
-        return file_name
-    else:
-        print("Failed to find the title tag")
-        return None
+    file_id = url.split('/')[-2]
+    gdrive_url = f'https://drive.google.com/uc?id={file_id}'
+    return gdown.download(url=gdrive_url, quiet=True)
 
 def get_name(url, gdrive):