VatsalPatel18 commited on
Commit
d73a21c
·
verified ·
1 Parent(s): a852a36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -53,10 +53,7 @@ if ZERO_GPU:
53
  user = (
54
  f"Brief: {prompt}\nBase: primary={base_primary}, secondary={base_secondary}, accent={base_accent}"
55
  )
56
- out = textgen(sys + "
57
- " + user, max_new_tokens=160, do_sample=True, temperature=0.6)[0][
58
- "generated_text"
59
- ]
60
  import re
61
  hexes = re.findall(r"#(?:[0-9a-fA-F]{3}){1,2}", out)
62
  note = "AI suggested a professional palette with good contrast."
 
53
  user = (
54
  f"Brief: {prompt}\nBase: primary={base_primary}, secondary={base_secondary}, accent={base_accent}"
55
  )
56
+ out = textgen(sys + "\n" + user, max_new_tokens=160, do_sample=True, temperature=0.6)[0]["generated_text"]
 
 
 
57
  import re
58
  hexes = re.findall(r"#(?:[0-9a-fA-F]{3}){1,2}", out)
59
  note = "AI suggested a professional palette with good contrast."