Update app.py

app.py

@@ -10,17 +10,15 @@ compressor = PromptCompressor(model_name="openai-community/gpt2", device_map="cp
 class TextInput(BaseModel):
     text: str
 
-class TextOutput(BaseModel):
-    compressed_text: str
-
-@app.post("/compress", response_model=TextOutput)
+@app.post("/compress")
 async def compress_text(input: TextInput):
     try:
         compressed_text = compressor.compress_prompt(input.text)
-        return TextOutput(compressed_text=compressed_text)
+        # Return raw data for now
+        return {"compressed_text": compressed_text}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.get("/")
 async def root():
-    return {"message": "Welcome to the LLMLingua Text Compression API"}
+    return {"message": "Welcome to the LLMLingua Text Compression API"}