Upload app.py

app.py

@@ -17,7 +17,7 @@ def compressit(original_text, compressor1, ratio, maxlength):
     elif compressor1=="SCRL":
         compressor = PromptCompressor(type='SCRLCompressor', model_dir="models/gigaword-L8/", device="cpu", tokenizer_dir="sentence-transformers/paraphrase-distilroberta-base-v2")
     elif compressor1=="KiS":
-        compressor = PromptCompressor(type='KiSCompressor', device="cuda", model_dir="philippelaban/keep_it_simple")
+        compressor = PromptCompressor(type='KiSCompressor', device="cpu", model_dir="philippelaban/keep_it_simple")
     else:
         compressor = PromptCompressor(type='SCCompressor', lang='en', model='gpt2', device='cpu')
 
@@ -39,7 +39,7 @@ demo = gr.Interface(
         gr.Textbox(lines=1, placeholder="Enter the max_length parameter if you are using SCRL or KiS", label="max_length", info="If you are using SCRL or KiS, fill in the parameter, if not, just ignore this.\n Hint: For SCRL, max_length should be shorter than the lenght of original prompt; For KiS, max_length should be longer than it.")
     ],
     outputs=[
-        gr.Textbox(lines=1, info="Please note that when the text is very short, LLMLingua and LongLLMLingua will not work.")
+        gr.Textbox(lines=1, info="Please wait patiently when proceeding it may take more than 2 minutes to generate since we are using CPUs for free.")
     ]
 )