Update app.py

app.py

@@ -3,7 +3,6 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from backtrack_sampler import BacktrackSampler, CreativeWritingStrategy
 from backtrack_sampler.provider.transformers_provider import TransformersProvider
 import torch
-import asyncio
 import spaces
 
 description = """## Compare Creative Writing: Custom Sampler vs. Backtrack Sampler with Creative Writing Strategy
@@ -53,7 +52,7 @@ def generate_responses(prompt, history):
             generated_list.append(token)
         return tokenizer.decode(generated_list, skip_special_tokens=True)
     standard_output = standard_task()
-    custom_output = asyncio.run(custom_sampler_task())
+    custom_output = custom_sampler_task()
     # Decode standard output and remove the prompt from the generated response
     standard_response = tokenizer.decode(standard_output[0][len(inputs[0]):], skip_special_tokens=True)