sdkrastev commited on
Commit
bacaeb4
·
verified ·
1 Parent(s): db4d8b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -3,13 +3,12 @@ import spaces
3
  from huggingface_hub import InferenceClient
4
  from transformers import pipeline
5
 
6
- LOCAL_MODEL = "microsoft/Phi-3-mini-4k-instruct"
7
  REMOTE_MODEL = "openai/gpt-oss-20b"
8
 
9
  pipe = pipeline(
10
  "text-generation",
11
  model=LOCAL_MODEL,
12
- trust_remote_code=True,
13
  dtype="auto",
14
  device="cuda",
15
  )
 
3
  from huggingface_hub import InferenceClient
4
  from transformers import pipeline
5
 
6
+ LOCAL_MODEL = "Qwen/Qwen3-0.6B"
7
  REMOTE_MODEL = "openai/gpt-oss-20b"
8
 
9
  pipe = pipeline(
10
  "text-generation",
11
  model=LOCAL_MODEL,
 
12
  dtype="auto",
13
  device="cuda",
14
  )