Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -355,7 +355,7 @@ if __name__ == "__main__":
|
|
355 |
parser = argparse.ArgumentParser()
|
356 |
parser.add_argument("--share", action="store_true", default=False, help="share gradio app")
|
357 |
parser.add_argument("--port", type=int, default=7860, help="gradio server port")
|
358 |
-
parser.add_argument("--device", type=str, default="
|
359 |
parser.add_argument("--batch", type=int, default=8, help="batch size")
|
360 |
parser.add_argument("--max-gen", type=int, default=1024, help="max")
|
361 |
opt = parser.parse_args()
|
|
|
355 |
parser = argparse.ArgumentParser()
|
356 |
parser.add_argument("--share", action="store_true", default=False, help="share gradio app")
|
357 |
parser.add_argument("--port", type=int, default=7860, help="gradio server port")
|
358 |
+
parser.add_argument("--device", type=str, default="cpu", help="device to run model")
|
359 |
parser.add_argument("--batch", type=int, default=8, help="batch size")
|
360 |
parser.add_argument("--max-gen", type=int, default=1024, help="max")
|
361 |
opt = parser.parse_args()
|