kgupta21 commited on
Commit
4a88ecb
·
1 Parent(s): 746ae2b

local inference page with fixes to gpu with zerogpu + add accelerate for device mapping - removed previous and fixed overall

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -341,4 +341,4 @@ with gr.Blocks() as demo:
341
  )
342
 
343
  logger.info("Starting Gradio interface...")
344
- demo.queue().launch(ssr_mode=False)
 
341
  )
342
 
343
  logger.info("Starting Gradio interface...")
344
+ demo.queue().launch()
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  transformers
2
- gradio
3
  torch
4
  accelerate
5
  SentencePiece
 
1
  transformers
2
+ gradio>=4.44.1
3
  torch
4
  accelerate
5
  SentencePiece