Test 2

app.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from transformers import AutoModelForCausalLM, AutoTokenizer
-from typing import List, Dict
+from typing import List
 import torch
 
 app = FastAPI(title="Language Model API")
@@ -50,8 +50,7 @@ async def generate_response(request: ChatRequest):
         response_text = tokenizer.decode(outputs[0])
         
         return {
-            "generated_text": response_text,
-            "input_text": input_text
+            "generated_text": response_text
         }
     
     except Exception as e:
@@ -59,4 +58,4 @@ async def generate_response(request: ChatRequest):
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)