NimaKL commited on
Commit
4447663
·
verified ·
1 Parent(s): ad3da90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -7,7 +7,6 @@ from torch_geometric.nn import GATConv
7
  from sentence_transformers import SentenceTransformer
8
  from sklearn.metrics.pairwise import cosine_similarity
9
 
10
- # Define the GATConv model architecture
11
  class ModeratelySimplifiedGATConvModel(torch.nn.Module):
12
  def __init__(self, in_channels, hidden_channels, out_channels):
13
  super().__init__()
 
7
  from sentence_transformers import SentenceTransformer
8
  from sklearn.metrics.pairwise import cosine_similarity
9
 
 
10
  class ModeratelySimplifiedGATConvModel(torch.nn.Module):
11
  def __init__(self, in_channels, hidden_channels, out_channels):
12
  super().__init__()