NadiaBedhiafi commited on
Commit
c8cf210
·
verified ·
1 Parent(s): 7637286

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from transformers import pipeline
10
  #client = InferenceClient("microsoft/Phi-3-mini-128k-instruct")
11
  #client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
12
  #client = InferenceClient("google/gemma-1.1-7b-it")
13
- pipe = pipeline("text-generation", model="segolilylabs/Lily-Cybersecurity-7B-v0.2")
14
 
15
 
16
  @spaces.GPU
 
10
  #client = InferenceClient("microsoft/Phi-3-mini-128k-instruct")
11
  #client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
12
  #client = InferenceClient("google/gemma-1.1-7b-it")
13
+ pipe = pipeline("text-generation", model="internlm/internlm2_5-7b-chat", trust_remote_code=True)
14
 
15
 
16
  @spaces.GPU