KLeedrug commited on
Commit
cbfb9df
1 Parent(s): 5f981ab

add comment & change model into pretrained to see if it works

Browse files
Files changed (1) hide show
  1. app.py +15 -2
app.py CHANGED
@@ -4,11 +4,24 @@
4
  Automatically generated by Colaboratory.
5
 
6
  Original file is located at
7
- https://colab.research.google.com/drive/1vts9oE-egwIQU1A2cyDGjopOeTSS_Kk7
8
  """
9
 
10
  from EMO_AI.all import *
11
- model = get_model(pretrained=False)
 
 
 
 
 
 
 
 
 
 
 
 
 
12
 
13
  import gradio as gr
14
 
 
4
  Automatically generated by Colaboratory.
5
 
6
  Original file is located at
7
+ https://colab.research.google.com/drive/lalalalalal_HAHAHHAH!@#$%^&
8
  """
9
 
10
  from EMO_AI.all import *
11
+
12
+ """
13
+ # should add kwargs in get_model(), time to update our lib
14
+ from pathlib import Path
15
+ # return a dict of options
16
+ def load_weight(f="emo_0.pt"):
17
+ p = Path(f)
18
+ if p.is_file():
19
+ return {"PATH": f, "inference_only": False}
20
+ return {"pretrained":False}
21
+ """
22
+ model = get_model("emo_0.pt", inference_only=False)
23
+
24
+ # stable: model = get_model(pretrained=False)
25
 
26
  import gradio as gr
27