cstr commited on
Commit
82deaf2
·
verified ·
1 Parent(s): 4813eed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -315,7 +315,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
315
  )
316
 
317
  # API endpoint for external access
318
- @demo.load(api_name="generate")
319
  def api_generate(message, model=None, image_data=None):
320
  """API endpoint for generating responses"""
321
  model_name = model or text_models[0][0]
@@ -389,5 +389,9 @@ def api_generate(message, model=None, image_data=None):
389
  except Exception as e:
390
  return {"error": f"Error generating response: {str(e)}"}
391
 
 
 
 
392
  if __name__ == "__main__":
393
- demo.launch()
 
 
315
  )
316
 
317
  # API endpoint for external access
318
+ @demo.queue()
319
  def api_generate(message, model=None, image_data=None):
320
  """API endpoint for generating responses"""
321
  model_name = model or text_models[0][0]
 
389
  except Exception as e:
390
  return {"error": f"Error generating response: {str(e)}"}
391
 
392
+ demo.queue()
393
+ demo.launch(share=False)
394
+
395
  if __name__ == "__main__":
396
+ # Remove or comment out demo.launch() here if you added it above
397
+ pass