CK42 commited on
Commit
e4fa88d
·
1 Parent(s): f04ccc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def get_metadata(model_id):
41
  except requests.exceptions.HTTPError:
42
  return None
43
 
44
- classifier = pipeline("text-classification", model="juliensimon/distilbert-amazon-shoe-reviews")
45
 
46
  def predict(review):
47
  prediction = classifier(review)
 
41
  except requests.exceptions.HTTPError:
42
  return None
43
 
44
+ classifier = pipeline("text-classification", model)
45
 
46
  def predict(review):
47
  prediction = classifier(review)