Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,3 @@
|
|
1 |
-
import spaces
|
2 |
import os
|
3 |
import torch
|
4 |
import torch.nn as nn
|
@@ -6,6 +5,7 @@ import pandas as pd
|
|
6 |
import torch.nn.functional as F
|
7 |
from lavis.models.protein_models.protein_function_opt import Blip2ProteinMistral
|
8 |
from lavis.models.base_model import FAPMConfig
|
|
|
9 |
import gradio as gr
|
10 |
# from esm_scripts.extract import run_demo
|
11 |
from esm import pretrained, FastaBatchedDataset
|
|
|
|
|
1 |
import os
|
2 |
import torch
|
3 |
import torch.nn as nn
|
|
|
5 |
import torch.nn.functional as F
|
6 |
from lavis.models.protein_models.protein_function_opt import Blip2ProteinMistral
|
7 |
from lavis.models.base_model import FAPMConfig
|
8 |
+
import spaces
|
9 |
import gradio as gr
|
10 |
# from esm_scripts.extract import run_demo
|
11 |
from esm import pretrained, FastaBatchedDataset
|