File size: 1,201 Bytes
2924719
0e067b4
dc155d4
0e067b4
 
 
76bc6e5
2924719
 
 
7bd100b
dc155d4
0e067b4
 
 
 
bd1db85
0e067b4
 
 
1513fb8
0e067b4
67e379c
0e067b4
 
 
6ab11e0
0e067b4
6ab11e0
1513fb8
2924719
1513fb8
2924719
 
 
dc155d4
0e067b4
2924719
1513fb8
d144ea2
0e067b4
 
 
 
 
 
dc155d4
2924719
dc155d4
0e067b4
 
 
 
 
 
330958f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
import os
import sys

# ---------------------------------------------------------
# Unsloth Studio environment
# ---------------------------------------------------------

os.environ.setdefault(
    "UNSLOTH_STUDIO_HOME",
    "/tmp/unsloth_studio",
)

os.makedirs(
    os.environ["UNSLOTH_STUDIO_HOME"],
    exist_ok=True,
)

# ---------------------------------------------------------
# Import the REAL Unsloth Studio FastAPI application
# ---------------------------------------------------------

from studio.backend.main import app as studio_app

# ---------------------------------------------------------
# Gradio
# ---------------------------------------------------------

import gradio as gr


with gr.Blocks(title="Unsloth Studio") as demo:

    gr.Markdown(
        """
        # 🦥 Unsloth Studio

        Full Unsloth Studio running inside a Hugging Face Space.
        """
    )


# ---------------------------------------------------------
# Mount Studio into Gradio
# ---------------------------------------------------------

demo.app = studio_app


if __name__ == "__main__":
    import uvicorn

    uvicorn.run(
        studio_app,
        host="0.0.0.0",
        port=7860,
    )