javrtg commited on
Commit
5b644a4
·
1 Parent(s): c33a05e
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -19,6 +19,9 @@ RAD2DEG = 180 / pi
19
  DEV = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20
  MODEL_DIR = Path("/models") # mounted from https://huggingface.co/javrtg/AnyCalib
21
 
 
 
 
22
 
23
  def get_model(model_id: str) -> AnyCalib:
24
  model = AnyCalib(model_id=None)
@@ -188,7 +191,7 @@ def build_demo() -> gr.Blocks:
188
  <a href="https://scholar.google.com/citations?user=j_sMzokAAAAJ">Javier Civera</a><br>
189
  I3A, University of Zaragoza
190
  </h3>
191
- <img width="60%" src="https://github.com/javrtg/AnyCalib/blob/main/assets/method_dark.png?raw=true">
192
  <h3>
193
  Camera calibration from a single perspective/edited/distorted image using a freely chosen camera model<br>
194
  <a href="https://github.com/javrtg/AnyCalib">Code</a>
 
19
  DEV = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20
  MODEL_DIR = Path("/models") # mounted from https://huggingface.co/javrtg/AnyCalib
21
 
22
+ ASSETS_DIR = Path(__file__).resolve().parent / "assets"
23
+ gr.set_static_paths(paths=[ASSETS_DIR])
24
+
25
 
26
  def get_model(model_id: str) -> AnyCalib:
27
  model = AnyCalib(model_id=None)
 
191
  <a href="https://scholar.google.com/citations?user=j_sMzokAAAAJ">Javier Civera</a><br>
192
  I3A, University of Zaragoza
193
  </h3>
194
+ <img width="60%" src="/gradio_api/file=assets/method_dark.png">
195
  <h3>
196
  Camera calibration from a single perspective/edited/distorted image using a freely chosen camera model<br>
197
  <a href="https://github.com/javrtg/AnyCalib">Code</a>