rararara9999 commited on
Commit
2dd3f29
·
verified ·
1 Parent(s): 559e08d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -9,7 +9,12 @@ subprocess.check_call(["pip", "install", "evaluate"])
9
  subprocess.check_call(["pip", "install", "scikit-learn"])
10
  subprocess.check_call(["pip", "install", "torchvision"])
11
 
12
- from transformers import AutoModelForImageClassification, AutoImageProcessor
 
 
 
 
 
13
  import torch
14
  import numpy as np
15
  from PIL import Image
 
9
  subprocess.check_call(["pip", "install", "scikit-learn"])
10
  subprocess.check_call(["pip", "install", "torchvision"])
11
 
12
+ # Load model directly
13
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
14
+
15
+ tokenizer = AutoTokenizer.from_pretrained("rararara9999/Model")
16
+ model = AutoModelForSequenceClassification.from_pretrained("rararara9999/Model")
17
+
18
  import torch
19
  import numpy as np
20
  from PIL import Image