Update app.py

app.py

@@ -10,11 +10,9 @@ model_path = 'THUDM/BPO'
 
 device = 'cuda'
 
-if torch.cuda.is_available():
-
-    tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True, add_prefix_space=True)
-    model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True, device_map=device, load_in_8bit=True)
-    model = model.eval()
+tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True, add_prefix_space=True)
+model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True, device_map=device, load_in_8bit=True)
+model = model.eval()
 
 
 DESCRIPTION = """This Space demonstrates model [BPO](https://huggingface.co/THUDM/BPO), which is built on LLaMA-2-7b-chat.
@@ -137,7 +135,7 @@ def reset_state():
 def update_textbox_from_dropdown(selected_example):
     return selected_example
 
-with gr.Blocks("sty.css") as demo:
+with gr.Blocks(css="sty.css") as demo:
     gr.HTML("""<h1 align="center">Prompt Preference Optimizer</h1>""")
 
     gr.Markdown(DESCRIPTION)