yaron123 commited on
Commit
6e96755
·
1 Parent(s): fd72ef6
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -434,8 +434,8 @@ pegasus_name = "google/pegasus-xsum"
434
  # precision data
435
 
436
  seq=512
437
- width=768
438
- height=768
439
  image_steps=8
440
  img_accu=0
441
 
@@ -631,9 +631,9 @@ def handle_generate(artist,song,genre,lyrics):
631
  pos_genre = re.sub(f'[{punctuation}]', '', re.sub("([ \t\n]){1,}", " ", genre)).upper().strip()
632
  pos_lyrics = re.sub(f'[{punctuation}]', '', re.sub("([ \t\n]){1,}", " ", lyrics)).lower().strip()
633
  pos_lyrics_sum = pos_lyrics if pos_lyrics == "" else summarize_text(pos_lyrics)
634
- neg = f"Textual Labeled Distorted Discontinuous Ugly Blurry Low-Quality Worst-Quality Low-Resolution Painted"
635
  q = "\""
636
- pos = f'Realistic Vivid Genuine Reasonable Highly-Detailed 4K { pos_genre } { pos_song }{ pos_lyrics_sum if pos_lyrics_sum == "" else " " + q + pos_lyrics_sum + q }'
637
 
638
  print(f"""
639
  Positive: {pos}
 
434
  # precision data
435
 
436
  seq=512
437
+ width=1536
438
+ height=1536
439
  image_steps=8
440
  img_accu=0
441
 
 
631
  pos_genre = re.sub(f'[{punctuation}]', '', re.sub("([ \t\n]){1,}", " ", genre)).upper().strip()
632
  pos_lyrics = re.sub(f'[{punctuation}]', '', re.sub("([ \t\n]){1,}", " ", lyrics)).lower().strip()
633
  pos_lyrics_sum = pos_lyrics if pos_lyrics == "" else summarize_text(pos_lyrics)
634
+ neg = f"Textual Labeled Distorted Discontinuous Ugly Blurry Low-Quality Worst-Quality Low-Resolution Painted Pastel Vivid Soft"
635
  q = "\""
636
+ pos = f'Rough Realistic Natural Genuine Reasonable Highly-Detailed { pos_genre } { pos_song }{ pos_lyrics_sum if pos_lyrics_sum == "" else " " + q + pos_lyrics_sum + q }'
637
 
638
  print(f"""
639
  Positive: {pos}