Spaces:
Sleeping
Sleeping
Commit
·
f63d6cc
1
Parent(s):
90c8b35
fix: shared=True
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ import gradio as gr
|
|
9 |
|
10 |
def main(
|
11 |
base_model="ise-uiuc/Magicoder-S-DS-6.7B",
|
12 |
-
|
13 |
port=80,
|
14 |
):
|
15 |
pipeline = transformers.pipeline(
|
@@ -67,7 +67,7 @@ def main(
|
|
67 |
],
|
68 |
title="Magicoder",
|
69 |
description="This is a playground for Magicoder-S-DS-6.7B! Follow us on Github: https://github.com/ise-uiuc/magicoder and Huggingface: https://huggingface.co/ise-uiuc.",
|
70 |
-
).queue().launch(
|
71 |
|
72 |
|
73 |
if __name__ == "__main__":
|
|
|
9 |
|
10 |
def main(
|
11 |
base_model="ise-uiuc/Magicoder-S-DS-6.7B",
|
12 |
+
device="cuda" if torch.cuda.is_available() else "cpu",
|
13 |
port=80,
|
14 |
):
|
15 |
pipeline = transformers.pipeline(
|
|
|
67 |
],
|
68 |
title="Magicoder",
|
69 |
description="This is a playground for Magicoder-S-DS-6.7B! Follow us on Github: https://github.com/ise-uiuc/magicoder and Huggingface: https://huggingface.co/ise-uiuc.",
|
70 |
+
).queue().launch(server_port=port)
|
71 |
|
72 |
|
73 |
if __name__ == "__main__":
|