Update app.py

app.py

@@ -102,7 +102,7 @@ def run_protgpt2(startsequence, length, repetitionPenalty, top_k_poolsize, max_s
     #device.reset()
     #print(gpu_usage())
     return sequences
-
+@ray.remote
 def run_alphafold(startsequence):
     print(gpu_usage())
     model_runners = {}
@@ -152,7 +152,7 @@ def update(inp):
     print("Running AF on", inp)
     startsequence = inp
     # run alphafold using ray
-    plddts = run_alphafold(startsequence)
+    plddts = ray.get(run_alphafold.remote(startsequence))
     print(plddts)
     x = np.arange(10)
     #plt.style.use(["seaborn-ticks", "seaborn-talk"])
@@ -398,4 +398,6 @@ with proteindream:
     #seqChoice.change(fn=update_seqs, inputs=seqChoice, outputs=chosenSeq)
     btn2.click(fn=update, inputs=chosenSeq, outputs=[mol, plot, meanpLDDT])
 
+ray.init(runtime_env={"working_dir": "."})
+
 proteindream.launch(share=False)