Topallaj Denis commited on
Commit
ef19c0b
·
1 Parent(s): 6caf496
Files changed (1) hide show
  1. main.py +15 -1
main.py CHANGED
@@ -18,12 +18,13 @@ import pydantic
18
  app = FastAPI()
19
 
20
  app.mount("/", StaticFiles(directory="static", html=True), name="static")
21
-
22
  tokenizer = T5Tokenizer.from_pretrained(
23
  "Rostlab/prot_t5_xl_half_uniref50-enc", do_lower_case=False, torch_dtype=torch.float16)
24
 
25
  model = T5EncoderModel.from_pretrained(
26
  "Rostlab/prot_t5_xl_half_uniref50-enc")
 
27
 
28
  class Item(pydantic.BaseModel):
29
  sequence: str
@@ -35,6 +36,18 @@ def index() -> FileResponse:
35
 
36
  @app.post("/predict")
37
  def predict(item: Item):
 
 
 
 
 
 
 
 
 
 
 
 
38
  endpointHandler = EndpointHandler()
39
  result = endpointHandler.predict({
40
  "inputs": {
@@ -42,6 +55,7 @@ def predict(item: Item):
42
  "smiles": item.smiles
43
  }
44
  })
 
45
 
46
  return result
47
 
 
18
  app = FastAPI()
19
 
20
  app.mount("/", StaticFiles(directory="static", html=True), name="static")
21
+ """
22
  tokenizer = T5Tokenizer.from_pretrained(
23
  "Rostlab/prot_t5_xl_half_uniref50-enc", do_lower_case=False, torch_dtype=torch.float16)
24
 
25
  model = T5EncoderModel.from_pretrained(
26
  "Rostlab/prot_t5_xl_half_uniref50-enc")
27
+ """
28
 
29
  class Item(pydantic.BaseModel):
30
  sequence: str
 
36
 
37
  @app.post("/predict")
38
  def predict(item: Item):
39
+
40
+ sequence = item.sequence
41
+ smiles = item.smiles
42
+
43
+ result = {
44
+ "Km": 0.0,
45
+ "Kcat": 0.0,
46
+ "Vmax": 0.0
47
+ }
48
+
49
+
50
+ """
51
  endpointHandler = EndpointHandler()
52
  result = endpointHandler.predict({
53
  "inputs": {
 
55
  "smiles": item.smiles
56
  }
57
  })
58
+ """
59
 
60
  return result
61