Spaces:
Sleeping
Sleeping
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- app.py +1 -1
- 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(
|
|
|
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
|