Yaron Koresh commited on
Commit
dac7ef4
·
verified ·
1 Parent(s): 209a22d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -66,13 +66,13 @@ def generate_random_string(length):
66
  return ''.join(random.choice(characters) for _ in range(length))
67
 
68
  @spaces.GPU
69
- def Piper(_do):
70
  return pipe(
71
  _do,
72
  height=480,
73
  width=480,
74
- negative_prompt="",
75
- num_inference_steps=100,
76
  guidance_scale=10
77
  )
78
 
@@ -83,6 +83,7 @@ def infer(prompt1,prompt2,prompt3,prompt4):
83
  prompt5 = ""
84
  else:
85
  prompt5 = " or " + " or ".join([translate(v,"english") for v in prompt4])
 
86
  _dont = f'Show TEXT or LOGO{prompt5}.'
87
 
88
  if prompt4 == None:
@@ -107,8 +108,8 @@ def infer(prompt1,prompt2,prompt3,prompt4):
107
  prompt3 = " ".join([translate(v,"english").upper() for v in prompt3]) + " event"
108
 
109
  _do = f'Show {prompt3}; Include {prompt2}; Focus on {prompt1}; Remove {prompt6}.'
110
- print(_do)
111
- image = Piper(_do).images[0].save(name)
112
  return name
113
 
114
  css="""
 
66
  return ''.join(random.choice(characters) for _ in range(length))
67
 
68
  @spaces.GPU
69
+ def Piper(_do,_dont):
70
  return pipe(
71
  _do,
72
  height=480,
73
  width=480,
74
+ negative_prompt=_dont,
75
+ num_inference_steps=50,
76
  guidance_scale=10
77
  )
78
 
 
83
  prompt5 = ""
84
  else:
85
  prompt5 = " or " + " or ".join([translate(v,"english") for v in prompt4])
86
+
87
  _dont = f'Show TEXT or LOGO{prompt5}.'
88
 
89
  if prompt4 == None:
 
108
  prompt3 = " ".join([translate(v,"english").upper() for v in prompt3]) + " event"
109
 
110
  _do = f'Show {prompt3}; Include {prompt2}; Focus on {prompt1}; Remove {prompt6}.'
111
+
112
+ image = Piper(_do,_dont).images[0].save(name)
113
  return name
114
 
115
  css="""