Yntec commited on
Commit
e864205
·
verified ·
1 Parent(s): 1b22623

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ async def infer(model_str, prompt, seed=1, timeout=inference_timeout):
54
  noise = ""
55
  kwargs["seed"] = seed
56
  task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn,
57
- prompt=f'{prompt} {noise}', **kwargs, token=HF_TOKEN))
58
  await asyncio.sleep(0)
59
  try:
60
  result = await asyncio.wait_for(task, timeout=timeout)
 
54
  noise = ""
55
  kwargs["seed"] = seed
56
  task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn,
57
+ prompt=f'{prompt} {noise}', **kwargs))
58
  await asyncio.sleep(0)
59
  try:
60
  result = await asyncio.wait_for(task, timeout=timeout)