NikhilJoson commited on
Commit
28bb0de
·
verified ·
1 Parent(s): cd7a35d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -230,8 +230,8 @@ def generate_image(prompt, seed=None, guidance=5, t2i_temperature=1.0, progress=
230
  torch.manual_seed(seed)
231
  torch.cuda.manual_seed(seed)
232
  np.random.seed(seed)
233
- width = 512
234
- height = 512
235
  parallel_size = 4
236
 
237
  with torch.no_grad():
@@ -278,10 +278,7 @@ def clear_chat(image):
278
 
279
 
280
  # Gradio interface
281
- css = '''
282
- .gradio-container {max-width: 960px !important}
283
- '''
284
- with gr.Blocks(css=css) as demo:
285
  gr.Markdown("# Janus Pro 7B - Unified Chat Interface")
286
 
287
  # State variables to maintain context
 
230
  torch.manual_seed(seed)
231
  torch.cuda.manual_seed(seed)
232
  np.random.seed(seed)
233
+ width = 384
234
+ height = 384
235
  parallel_size = 4
236
 
237
  with torch.no_grad():
 
278
 
279
 
280
  # Gradio interface
281
+ with gr.Blocks() as demo:
 
 
 
282
  gr.Markdown("# Janus Pro 7B - Unified Chat Interface")
283
 
284
  # State variables to maintain context