Hugging Face's logo

Spaces:
Haus0226
/
Thesis
Sleeping

App Files Community
Haus0226 commited on
Commit
81b89bf
·
verified ·
1 Parent(s): 86bb2ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
  from PIL import Image
6
 
7
  set_seed(226)
8
- model = GitForCausalLM.from_pretrained("microsoft/git-base").to("cuda").eval()
9
  processor = GitProcessor.from_pretrained("microsoft/git-base")
10
 
11
  st.title("What is hidden ?")
 
5
  from PIL import Image
6
 
7
  set_seed(226)
8
+ model = GitForCausalLM.from_pretrained("microsoft/git-base").eval()
9
  processor = GitProcessor.from_pretrained("microsoft/git-base")
10
 
11
  st.title("What is hidden ?")