AndersonConforto commited on
Commit
b4ca481
·
1 Parent(s): 80be522

first commit

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +15 -0
app.py CHANGED
@@ -54,7 +54,7 @@ model = HelioSpectFormer(
54
 
55
  # Carregar pesos
56
  state_dict = torch.load(MODEL_FILE, map_location=torch.device('cpu'))
57
- model.load_state_dict(state_dict)
58
  model.eval()
59
 
60
  # ================================
 
54
 
55
  # Carregar pesos
56
  state_dict = torch.load(MODEL_FILE, map_location=torch.device('cpu'))
57
+ model.load_state_dict(state_dict, strict=False)
58
  model.eval()
59
 
60
  # ================================
requirements.txt CHANGED
@@ -6,3 +6,18 @@ matplotlib
6
  einops
7
  gradio
8
  requests
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  einops
7
  gradio
8
  requests
9
+ packaging
10
+ PyYAML
11
+
12
+ # Data / datasets used in surya.datasets.*
13
+ xarray
14
+ scikit-image
15
+ pandas
16
+ numba
17
+ hdf5plugin
18
+
19
+ # Logging/experiment tracking (used by surya.utils.log)
20
+ wandb
21
+
22
+ # Optional: only needed if you use ephemeris utilities in datasets
23
+ # sunpy