dalexanderch commited on
Commit
77d9187
·
1 Parent(s): 68d6e94

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  from glycowork.ml.processing import dataset_to_dataloader
3
  import numpy as np
4
  import torch
 
5
 
6
  def fn(model, class_list):
7
  def f(glycan):
@@ -29,6 +30,7 @@ class_list=['Amoebozoa', 'Animalia', 'Bacteria', 'Bamfordvirae', 'Chromista', 'E
29
  'Orthornavirae', 'Pararnavirae', 'Plantae', 'Proteoarchaeota', 'Protista', 'Riboviria']
30
 
31
  f = fn(model, class_list)
 
32
 
33
  demo = gr.Interface(
34
  fn=f,
 
2
  from glycowork.ml.processing import dataset_to_dataloader
3
  import numpy as np
4
  import torch
5
+ import os
6
 
7
  def fn(model, class_list):
8
  def f(glycan):
 
30
  'Orthornavirae', 'Pararnavirae', 'Plantae', 'Proteoarchaeota', 'Protista', 'Riboviria']
31
 
32
  f = fn(model, class_list)
33
+ os.system("pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.12.0+cpu.html")
34
 
35
  demo = gr.Interface(
36
  fn=f,