Files changed (1) hide show
  1. app.py +1 -7
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