multimodalart HF Staff commited on
Commit
53801f5
·
verified ·
1 Parent(s): 4310b9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -14,7 +14,6 @@ TRUSTED_UPLOADERS = ["KappaNeuro", "CiroN2022", "multimodalart", "Norod78", "joa
14
 
15
  def get_json_data(url):
16
  url_split = url.split('/')
17
- print("url_split: ", url_split)
18
  api_url = f"https://civitai.com/api/v1/models/{url_split[4]}"
19
  try:
20
  response = requests.get(api_url)
@@ -37,12 +36,10 @@ def check_nsfw(json_data, profile):
37
  return True
38
 
39
  def get_prompts_from_image(image_id):
40
- print("image_id: ", image_id)
41
  url = f'https://civitai.com/images/{image_id}'
42
  response = requests.get(url)
43
  soup = BeautifulSoup(response.text, 'html.parser')
44
  content = soup.find_all(class_='mantine-Code-root mantine-Code-block mantine-2v44jn')
45
- print(content)
46
  if(content):
47
  if(len(content) > 1):
48
  return content[0].text, content[1].text
@@ -246,7 +243,6 @@ For more details, including weighting, merging and fusing LoRAs, check the [docu
246
  # elif index > 1:
247
  # content += f"\n![Image {index}]({image})\n> {prompt}\n"
248
  readme_content += content + "\n"
249
- print(readme_content)
250
  with open(f"{folder}/README.md", "w") as file:
251
  file.write(readme_content)
252
 
@@ -368,7 +364,7 @@ def bulk_upload(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuthToken,
368
  if(url):
369
  try:
370
  upload_result = upload_civit_to_hf(profile, oauth_token, url, link_civit)
371
- gr.Warning(f"Model {link_civit} uploaded")
372
  yield upload_result
373
  except Exception as e:
374
  print(e)
 
14
 
15
  def get_json_data(url):
16
  url_split = url.split('/')
 
17
  api_url = f"https://civitai.com/api/v1/models/{url_split[4]}"
18
  try:
19
  response = requests.get(api_url)
 
36
  return True
37
 
38
  def get_prompts_from_image(image_id):
 
39
  url = f'https://civitai.com/images/{image_id}'
40
  response = requests.get(url)
41
  soup = BeautifulSoup(response.text, 'html.parser')
42
  content = soup.find_all(class_='mantine-Code-root mantine-Code-block mantine-2v44jn')
 
43
  if(content):
44
  if(len(content) > 1):
45
  return content[0].text, content[1].text
 
243
  # elif index > 1:
244
  # content += f"\n![Image {index}]({image})\n> {prompt}\n"
245
  readme_content += content + "\n"
 
246
  with open(f"{folder}/README.md", "w") as file:
247
  file.write(readme_content)
248
 
 
364
  if(url):
365
  try:
366
  upload_result = upload_civit_to_hf(profile, oauth_token, url, link_civit)
367
+ gr.Info(f"Model {link_civit} uploaded")
368
  yield upload_result
369
  except Exception as e:
370
  print(e)