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
|