Spaces:
Sleeping
Sleeping
theme removed
Browse files
app.py
CHANGED
@@ -6,10 +6,6 @@ import torch
|
|
6 |
import spaces
|
7 |
import os
|
8 |
|
9 |
-
theme = gr.themes.Base(
|
10 |
-
font=[gr.themes.GoogleFont('Libre Franklin'), gr.themes.GoogleFont('Public Sans'), 'system-ui', 'sans-serif'],
|
11 |
-
)
|
12 |
-
|
13 |
model_name1 = "microsoft/Phi-3-medium-4k-instruct"
|
14 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
15 |
model1 = AutoModelForCausalLM.from_pretrained(model_name1, device_map='cuda', torch_dtype=torch.float16, trust_remote_code=True)
|
@@ -111,9 +107,7 @@ with gr.Blocks() as max:
|
|
111 |
]
|
112 |
)
|
113 |
|
114 |
-
|
115 |
-
|
116 |
-
with gr.Blocks(theme=theme, title="Phi 3 Medium DEMO") as demo:
|
117 |
gr.Markdown("# Phi3 Medium all in one")
|
118 |
gr.TabbedInterface([max, min], ['Phi3 medium 128k','Phi3 medium 4k'])
|
119 |
|
|
|
6 |
import spaces
|
7 |
import os
|
8 |
|
|
|
|
|
|
|
|
|
9 |
model_name1 = "microsoft/Phi-3-medium-4k-instruct"
|
10 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
11 |
model1 = AutoModelForCausalLM.from_pretrained(model_name1, device_map='cuda', torch_dtype=torch.float16, trust_remote_code=True)
|
|
|
107 |
]
|
108 |
)
|
109 |
|
110 |
+
with gr.Blocks(title="Phi 3 Medium DEMO") as demo:
|
|
|
|
|
111 |
gr.Markdown("# Phi3 Medium all in one")
|
112 |
gr.TabbedInterface([max, min], ['Phi3 medium 128k','Phi3 medium 4k'])
|
113 |
|