dpassov commited on
Commit
238e66b
β€’
1 Parent(s): daea6c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  # Use a pipeline as a high-level helper
2
  from transformers import pipeline
3
 
4
- my_pipe = pipeline("text-generation", model="TheBloke/Yarn-Mistral-7B-128k-AWQ")
5
 
6
  def get_class_1(input):
7
  return my_pipe(input)
 
1
  # Use a pipeline as a high-level helper
2
  from transformers import pipeline
3
 
4
+ my_pipe = pipeline("text-generation", model="TheBloke/Yarn-Mistral-7B-128k-GPTQ")
5
 
6
  def get_class_1(input):
7
  return my_pipe(input)