Nymbo commited on
Commit
8a00879
1 Parent(s): 7d541c4

converting to standard GPUs

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,7 +3,7 @@ import random
3
 
4
  import gradio as gr
5
  import numpy as np
6
- import spaces
7
  import torch
8
  from diffusers import DiffusionPipeline, LCMScheduler
9
 
@@ -50,7 +50,7 @@ def update_selection(
50
  return lora_id, trigger_word
51
 
52
 
53
- @spaces.GPU
54
  def infer(
55
  pre_prompt,
56
  prompt,
@@ -138,7 +138,7 @@ with gr.Blocks(css=css) as demo:
138
 
139
  with gr.Row():
140
 
141
- with gr.Blocks():
142
 
143
  with gr.Column():
144
 
 
3
 
4
  import gradio as gr
5
  import numpy as np
6
+ # import spaces
7
  import torch
8
  from diffusers import DiffusionPipeline, LCMScheduler
9
 
 
50
  return lora_id, trigger_word
51
 
52
 
53
+ # @spaces.GPU
54
  def infer(
55
  pre_prompt,
56
  prompt,
 
138
 
139
  with gr.Row():
140
 
141
+ with gr.Blocks(theme="Nymbo/Nymbo_Theme"):
142
 
143
  with gr.Column():
144