Update app.py

app.py

@@ -12,13 +12,17 @@ examples = [
     ["GSHMSDNEDNFDGDDFDDVEEDEGLDDLENAEEEGQENVEILPSGERPQANQKRITTPYMTKYERARVLGTRALQIAMCAPVMVELEGETDPLLIAMKELKARKIPIIIRRYLPDGSYEDWGVDELIITD"]]
 
 def get_out(sent):
+    prefix = ""
+    if len(sent)>1026:
+        sent =  sent[:1026]
+        prefix = "Your protein was longer than 1026 AAs. We are working on including longer sequences but in the meantime, here are the scores for the first 1026 AAs: "
     encoded = tokenizer.encode_plus(sent, return_tensors="pt")
     with torch.no_grad():
         output = model(**encoded)
     output = F.softmax(torch.squeeze(output['logits']))[1:-1,1].detach().numpy()
     output = np.array2string(output, precision=4, separator=',',
                       suppress_small=True)
-    return output
+    return prefix+output
 
 gr.Interface(
     get_out,