Update app.py

app.py

@@ -16,14 +16,14 @@ tokenizer = AutoTokenizer.from_pretrained(model_checkpoint1)
 class MyModel(nn.Module):
     def __init__(self):
         super().__init__()
-        self.bert1 = EsmForSequenceClassification.from_pretrained(model_checkpoint1, num_labels=3000)  # 3000
+        self.bert1 = EsmForSequenceClassification.from_pretrained(model_checkpoint1, num_labels=512)  # 3000
         # for param in self.bert1.parameters():
         #     param.requires_grad = False
         self.bn1 = nn.BatchNorm1d(256)
         self.bn2 = nn.BatchNorm1d(128)
         self.bn3 = nn.BatchNorm1d(64)
         self.relu = nn.LeakyReLU()
-        self.fc1 = nn.Linear(3000, 256)
+        self.fc1 = nn.Linear(512, 256)
         self.fc2 = nn.Linear(256, 128)
         self.fc3 = nn.Linear(128, 64)
         self.output_layer = nn.Linear(64, 2)
@@ -96,7 +96,7 @@ if __name__ == "__main__":
             "<p align='center' style='font-size: 20px;'>🔥Welcome to Antimicrobial Peptide Recognition Model. See our <a href='https://github.com/wrab12/diff-amp'>Project</a></p>")
         gr.HTML(
             '''<center>
-  <a href="https://huggingface.co/spaces/jackrui/diff-amp-AMP_Sequence_Detector?duplicate=true">
+  <a href="https://huggingface.co/spaces/jackrui/ampD?duplicate=true">
     <img src="https://bit.ly/3gLdBN6" alt="Duplicate Space">
   </a>
 </center>''')
@@ -131,7 +131,7 @@ if __name__ == "__main__":
         gr.Markdown(
             "<p align='center'><img src='https://pic4.zhimg.com/v2-eb2a7c0e746e67d1768090eec74f6787_b.jpg'></p>")
         gr.Markdown(
-            "<p align='center' style='font-size: 20px;'>Related job links in the same series: <a href='https://huggingface.co/spaces/jackrui/diff-amp-antimicrobial_peptide_generation'>diff-amp-antimicrobial_peptide_generation</a> , <a href='https://huggingface.co/spaces/jackrui/Diff-AMP-property-prediction-model'>Diff-AMP-property-prediction-model</a></p>")
+            "<p align='center' style='font-size: 20px;'>Related job links in the same series: <a href='https://huggingface.co/spaces/jackrui/ampG'>diff-amp-antimicrobial_peptide_generation</a> , <a href='https://huggingface.co/spaces/jackrui/ampPP'>Diff-AMP-property-prediction-model</a></p>")
         gr.Markdown('''📝 **Citation**
 If our work is useful for your research, please consider citing:
 ```
@@ -158,8 +158,7 @@ td {
     
 </tr>
 </table>
-
-<center><img src='https://api.infinitescript.com/badgen/count?name=jackrui/diff-amp-AMP_Sequence_Detector&ltext=Visitors&color=6dc9aa' alt='visitors'></center>
+<center><img src='https://api.infinitescript.com/badgen/count?name=jackrui/ampD&ltext=Visitors&color=6dc9aa' alt='visitors'></center>
 """
 ''')