Update main.py

main.py

@@ -11,6 +11,7 @@ from llmlingua import PromptCompressor
 llm_lingua = PromptCompressor(
     model_name="microsoft/llmlingua-2-bert-base-multilingual-cased-meetingbank",
     use_llmlingua2=True, # Whether to use llmlingua-2
+    device_map="cpu"
 )