Upload app.py

app.py

@@ -30,7 +30,7 @@ class PDBSearchAssistant:
             max_new_tokens=1024,
             temperature=0.1,
             torch_dtype="auto",
-            device="cuda"
+            device="mps"
         )
         
         self.prompt_template = """
@@ -71,6 +71,15 @@ class PDBSearchAssistant:
             Organism: Homo sapiens
             Similarity: none
 
+            Query: "mouse insulin"
+            Protein: insulin
+            Resolution: none
+            Sequence: none
+            PDB_ID: none
+            Method: none
+            Organism: Mus musculus
+            Similarity: none
+            
             Query: "Spike protein"
             Protein: Spike protein
             Resolution: none