Hugging Face's logo

Spaces:
runaksh
/
Symptom-2-disease_Text_Classification
Runtime error

App Files Community
runaksh commited on
Commit
617205e
·
1 Parent(s): ecd2d5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from PIL import Image
8
  from timeit import default_timer as timer
9
  from tensorflow import keras
10
  import torch
11
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
12
  import numpy as np
13
 
14
  loaded_model = AutoModelForSequenceClassification.from_pretrained("runaksh/Symptom-2-disease_distilBERT")
 
8
  from timeit import default_timer as timer
9
  from tensorflow import keras
10
  import torch
11
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
12
  import numpy as np
13
 
14
  loaded_model = AutoModelForSequenceClassification.from_pretrained("runaksh/Symptom-2-disease_distilBERT")