Polo123 commited on
Commit
31ac8a4
1 Parent(s): a20a100

Create logic.py

Browse files
Files changed (1) hide show
  1. logic.py +10 -0
logic.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import UdopProcessor, UdopForConditionalGeneration
2
+
3
+
4
+ def load_Model():
5
+ repo_id = "microsoft/udop-large"
6
+ processor = UdopProcessor.from_pretrained(repo_id)
7
+ model = UdopForConditionalGeneration.from_pretrained(repo_id)
8
+
9
+ return model, processor
10
+