xeroISB commited on
Commit
2b7acb8
·
verified ·
1 Parent(s): a96f819

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -8
README.md CHANGED
@@ -12,20 +12,17 @@ This is a model for predicting issue resolution times based on various features.
12
  ## How to use
13
 
14
  ```python
15
- import transformers as tf
16
- from keras.models import load_model
17
 
18
- # Load the model
19
- model = load_model("my_model.h5")
20
-
21
- # Prepare your input data (example)
22
- # X = ...
23
 
24
  # Predict
25
  predictions = model.predict(X)
26
  Model Details
27
  ```
28
- Framework: TensorFlow / Keras / Bert
29
  Input features: Text and numerical features
30
  Output: Duration category (short, medium, long)
31
  Performance
 
12
  ## How to use
13
 
14
  ```python
15
+ # Load model directly
16
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
17
 
18
+ tokenizer = AutoTokenizer.from_pretrained("xeroISB/ServiceNowMTTR")
19
+ model = AutoModelForSequenceClassification.from_pretrained("xeroISB/ServiceNowMTTR")
 
 
 
20
 
21
  # Predict
22
  predictions = model.predict(X)
23
  Model Details
24
  ```
25
+ Framework: Transformers / Bert
26
  Input features: Text and numerical features
27
  Output: Duration category (short, medium, long)
28
  Performance