Alesteba commited on
Commit
650cfa5
1 Parent(s): d2f871b

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 AutoModelForSequenceClassification
10
  # model = from_pretrained_keras("Alesteba/deep_model_07")
11
 
12
  from huggingface_hub import from_pretrained_fastai
13
- learner = from_pretrained_fastai("Alesteba/deep_model_07")
14
 
15
  def predict(prompt):
16
  return model.predict(prompt)[0]
 
10
  # model = from_pretrained_keras("Alesteba/deep_model_07")
11
 
12
  from huggingface_hub import from_pretrained_fastai
13
+ model = from_pretrained_fastai("Alesteba/deep_model_07")
14
 
15
  def predict(prompt):
16
  return model.predict(prompt)[0]