Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -62,7 +62,8 @@ metalatte_path = '.'
|
|
| 62 |
sys.path.insert(0, metalatte_path)
|
| 63 |
from configuration import MetaLATTEConfig
|
| 64 |
from modeling_metalatte import MultitaskProteinModel
|
| 65 |
-
|
|
|
|
| 66 |
# Load config + instantiate model (no network)
|
| 67 |
config = AutoConfig.from_pretrained(metalatte_local, local_files_only=True)
|
| 68 |
|
|
|
|
| 62 |
sys.path.insert(0, metalatte_path)
|
| 63 |
from configuration import MetaLATTEConfig
|
| 64 |
from modeling_metalatte import MultitaskProteinModel
|
| 65 |
+
AutoConfig.register("metalatte", MetaLATTEConfig)
|
| 66 |
+
AutoModel.register(MetaLATTEConfig, MultitaskProteinModel)
|
| 67 |
# Load config + instantiate model (no network)
|
| 68 |
config = AutoConfig.from_pretrained(metalatte_local, local_files_only=True)
|
| 69 |
|