karthik18AI commited on
Commit
e86825c
·
verified ·
1 Parent(s): 8ef2b9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,9 +1,8 @@
1
-
2
  # Use a pipeline as a high-level helper
3
  from transformers import pipeline
4
 
5
  messages = [
6
  {"role": "user", "content": "Who are you?"},
7
  ]
8
- pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
9
  pipe(messages)
 
 
1
  # Use a pipeline as a high-level helper
2
  from transformers import pipeline
3
 
4
  messages = [
5
  {"role": "user", "content": "Who are you?"},
6
  ]
7
+ pipe = pipeline("text-generation", model="Mxode/minicoder-7M-init")
8
  pipe(messages)