HardbanRecordsLab commited on
Commit
1980206
·
verified ·
1 Parent(s): 2f87114

Create inference.py

Browse files
Files changed (1) hide show
  1. inference.py +23 -0
inference.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+
4
+ ### 6. `inference.py`
5
+
6
+ ```python
7
+ import torch
8
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
9
+
10
+ def main():
11
+ model_name = "hardbanai-model"
12
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
13
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
14
+
15
+ text = input("Wpisz tekst: ")
16
+ inputs = tokenizer(text, return_tensors="pt")
17
+
18
+ outputs = model.generate(**inputs)
19
+ result = tokenizer.decode(outputs[0], skip_special_tokens=True)
20
+ print("Wynik:", result)
21
+
22
+ if __name__ == "__main__":
23
+ main()