JCai commited on
Commit
5952a1b
1 Parent(s): 3ee559d

change model

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -7,15 +7,15 @@ from typing import Iterable
7
  from gradio.themes.base import Base
8
  from gradio.themes.utils import colors, fonts, sizes
9
 
10
- import os
11
- from dotenv import load_dotenv
12
- load_dotenv()
13
-
14
- HF_ACCESS = os.getenv("HF_ACCESS")
15
 
16
  # Inference client setup
17
  client = InferenceClient(model="mistralai/Mistral-Small-Instruct-2409",
18
- token=HF_ACCESS
19
  )
20
  pipe = pipeline("text-generation", "microsoft/Phi-3-mini-4k-instruct", torch_dtype=torch.bfloat16, device_map="auto")
21
 
 
7
  from gradio.themes.base import Base
8
  from gradio.themes.utils import colors, fonts, sizes
9
 
10
+ # import os
11
+ # from dotenv import load_dotenv
12
+ # load_dotenv()
13
+ #
14
+ # HF_ACCESS = os.getenv("HF_ACCESS")
15
 
16
  # Inference client setup
17
  client = InferenceClient(model="mistralai/Mistral-Small-Instruct-2409",
18
+ # token=HF_ACCESS
19
  )
20
  pipe = pipeline("text-generation", "microsoft/Phi-3-mini-4k-instruct", torch_dtype=torch.bfloat16, device_map="auto")
21