Change from kmer 6 to kmer 4

app.py

@@ -26,7 +26,7 @@ class VirusClassifier(nn.Module):
     def forward(self, x):
         return self.network(x)
 
-def sequence_to_kmer_vector(sequence: str, k: int = 6) -> np.ndarray:
+def sequence_to_kmer_vector(sequence: str, k: int = 4) -> np.ndarray:
     """Convert sequence to k-mer frequency vector"""
     kmers = [''.join(p) for p in product("ACGT", repeat=k)]
     kmer_dict = {kmer: 0 for kmer in kmers}
@@ -64,10 +64,10 @@ def parse_fasta(fasta_content: str):
 def predict_sequence(fasta_content: str) -> str:
     """Process FASTA input and return formatted predictions"""
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
-    k = 6
+    k = 4
     
     # Load model and scaler
-    model = VirusClassifier(4096).to(device)  # 4096 = 4^6 for 6-mers
+    model = VirusClassifier(256).to(device)  # 256 = 4^4 for 4-mers
     model.load_state_dict(torch.load('model.pt', map_location=device))
     scaler = joblib.load('scaler.pkl')
     model.eval()