Update selective_context_compressor.py

selective_context_compressor.py

@@ -27,17 +27,3 @@ class SCCompressor(AbstractCompressor):
         }
         return result
 
-
-if __name__ == '__main__':
-    import time
-
-    compressor = SCCompressor(lang='en', model='gpt2', device='cuda')
-    test_prompt = "You belong to good side. In reveal phase, You can know which two players are Morgana and Assassin but you can't know which one is Morgana or Assassin specifically, you should reason it by yourself as the game progresses."
-
-    # level can be ['sent', 'phrase', 'token']; model can be ['gpt2', 'curie'], lang can only be 'en'
-    # if model is 'curie', an OPENAI-API-Key must be given in selective_context.py
-    start = time.time()
-    result = compressor.compress(original_prompt=test_prompt, ratio=0.4, level='phrase')
-    end = time.time()
-    print(result)
-    print('程序运行时间为: %s Seconds' % (end - start))