Hugging Face's logo

Spaces:
JasonTPhillipsJr
/
SpaGAN
Sleeping

App Files Community
JasonTPhillipsJr commited on
Commit
01b8a52
·
verified ·
1 Parent(s): fea8f2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  import spacy
3
  import torch
4
  import torch.nn as nn
5
- from transformers import BertTokenizer, BertModel
6
  from transformers.models.bert.modeling_bert import BertForMaskedLM
7
 
8
  from models.spabert.models.spatial_bert_model import SpatialBertConfig, SpatialBertForMaskedLM, SpatialBertModel
 
2
  import spacy
3
  import torch
4
  import torch.nn as nn
5
+ from transformers import BertTokenizer, BertModel, AutoConfig
6
  from transformers.models.bert.modeling_bert import BertForMaskedLM
7
 
8
  from models.spabert.models.spatial_bert_model import SpatialBertConfig, SpatialBertForMaskedLM, SpatialBertModel