AlekseyCalvin commited on
Commit
a1cb868
·
verified ·
1 Parent(s): 465756d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ if not hf_token:
21
  raise ValueError("Hugging Face API token not found. Please set the 'waffles' environment variable.")
22
 
23
 
24
- pipe = DiffusionPipeline.from_pretrained("models/AlekseyCalvin/RCA_Agitprop_Manufactory", torch_dtype=dtype, token=hf_token).to(device)
25
 
26
  @spaces.GPU()
27
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
 
21
  raise ValueError("Hugging Face API token not found. Please set the 'waffles' environment variable.")
22
 
23
 
24
+ pipe = DiffusionPipeline.from_pretrained("AlekseyCalvin/RCA_Agitprop_Manufactory", torch_dtype=dtype, token=hf_token).to(device)
25
 
26
  @spaces.GPU()
27
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):