Update app.py

app.py

@@ -46,13 +46,15 @@ class TextData(BaseModel):
     text: str
 
 # Set up logging
-logging.basicConfig(filename="logs_value.log", level=logging.INFO)
-
+def log_text(text: str):
+    timestamp = datetime.now().isoformat()
+    log_entry = f"{timestamp} - {text}\n"
+    with open("text_logs.txt", "a") as file:
+        file.write(log_entry)
+        
 @app.post("/predict")
 async def predict(data: TextData):
-    timestamp = datetime.now().isoformat()
-    log_entry = f"{timestamp} - {data.text}"
-    logging.info(log_entry)
+    log_text(data.text)
     inputs = tokenizer(data.text, return_tensors="pt", padding=True, truncation=True)
     inputs = {k: v.to(device1) for k, v in inputs.items()}
     
@@ -70,9 +72,7 @@ async def predict(data: TextData):
 
 @app.post("/predict_v2")
 async def predict_v2(data: TextData):
-    timestamp = datetime.now().isoformat()
-    log_entry = f"{timestamp} - {data.text}"
-    logging.info(log_entry)
+    log_text(data.text)
     
     inputs = tokenizer(data.text, return_tensors="pt", padding=True, truncation=True)
     inputs = {k: v.to(device2) for k, v in inputs.items()}