Update app.py

app.py

@@ -21,7 +21,7 @@ import gc
 import jax
 from numba import cuda
 print('GPU available',torch.cuda.is_available())
-print('__CUDA Device Name:',torch.cuda.get_device_name(0))
+#print('__CUDA Device Name:',torch.cuda.get_device_name(0))
 print(os.getcwd())
 if "/home/user/app/alphafold" not in sys.path:
     sys.path.append("/home/user/app/alphafold")
@@ -81,7 +81,7 @@ def predict_structure(prefix, feature_dict, model_runners, random_seed=0):
             f.write(protein.to_pdb(unrelaxed_protein))
     return plddts
 
-@ray.remote(num_gpus=1, max_calls=1)
+@ray.remote(num_gpus=0, max_calls=1)
 def run_protgpt2(startsequence, length, repetitionPenalty, top_k_poolsize, max_seqs):
     print("running protgpt2")
     print(gpu_usage())
@@ -103,7 +103,7 @@ def run_protgpt2(startsequence, length, repetitionPenalty, top_k_poolsize, max_s
     #print(gpu_usage())
     return sequences
     
-@ray.remote(num_gpus=1, max_calls=1)
+@ray.remote(num_gpus=0, max_calls=1)
 def run_alphafold(startsequence):
     print(gpu_usage())
     model_runners = {}