Update app.py

app.py

@@ -30,7 +30,7 @@ class PDBSearchAssistant:
             max_new_tokens=1024,
             temperature=0.1,
             torch_dtype="auto",
-            device="gpu"
+            device="cuda"
         )
         
         self.prompt_template = """