alomari7 commited on
Commit
f46b4ed
·
verified ·
1 Parent(s): 0d67961

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -54,7 +54,8 @@ def infer(prompt, negative_prompt, guidance_scale, num_inference_steps, seed, ra
54
  if randomize_seed:
55
  seed = random.randint(0, MAX_SEED)
56
 
57
- generator = torch.Generator(device=DEVICE).manual_seed(seed)
 
58
 
59
  # Update UI to show loading state
60
  yield {
@@ -125,7 +126,7 @@ with gr.Blocks(theme=theme, css="#footer {text-align: center;}") as demo:
125
  <h4><b>Development Team:</b></h4>
126
  <ul>
127
  <li>Osama Saeed</li>
128
- <li>Tariq Al-Amri</li>
129
  </ul>
130
  <hr>
131
  <h4><b>Special Thanks:</b></h4>
 
54
  if randomize_seed:
55
  seed = random.randint(0, MAX_SEED)
56
 
57
+ generator = torch.Generator
58
+ (device=DEVICE).manual_seed(seed)
59
 
60
  # Update UI to show loading state
61
  yield {
 
126
  <h4><b>Development Team:</b></h4>
127
  <ul>
128
  <li>Osama Saeed</li>
129
+ <li>Tareq Al-Omari</li>
130
  </ul>
131
  <hr>
132
  <h4><b>Special Thanks:</b></h4>