Ravi21 commited on
Commit
5359b48
·
verified ·
1 Parent(s): 97ba3f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -429,7 +429,7 @@ core=pv.Core()
429
  import ipywidgets as widgets
430
  device=widgets.Dropdown(
431
  options=core.available_devices+["AUTO"],
432
- value="CPU",
433
  desciption="Device:",
434
  disabled=False,
435
  )
@@ -463,6 +463,8 @@ seed=widgets.IntSlider(min=0,max=10000000,description="seed:",value=54)
463
  widgets.VBox([text_prompt,seed,num_steps])
464
  result=pv_pipe(text_prompt.value,num_inference_steps=num_steps.value,seed=seed.value)
465
 
 
 
466
  import gradio as gr
467
 
468
  def generate_text(text,seed,num_steps,strength,_=gr.Progress(track_tqdm=True)):
 
429
  import ipywidgets as widgets
430
  device=widgets.Dropdown(
431
  options=core.available_devices+["AUTO"],
432
+ value="GPU",
433
  desciption="Device:",
434
  disabled=False,
435
  )
 
463
  widgets.VBox([text_prompt,seed,num_steps])
464
  result=pv_pipe(text_prompt.value,num_inference_steps=num_steps.value,seed=seed.value)
465
 
466
+
467
+ @spaces.GPU(enable_queue=True)
468
  import gradio as gr
469
 
470
  def generate_text(text,seed,num_steps,strength,_=gr.Progress(track_tqdm=True)):