Add logging and print statements for debugging

app.py

@@ -29,6 +29,7 @@ def check_thread(logging_queue: Queue):
     print("Logging thread started.")
     print(f"Logging to '{getenv('OUTPUT_DATASET')}'")
     while True:
+        print("Checking for logs...")
         sleep(60)
         batch = []
         while not logging_queue.empty():
@@ -72,7 +73,6 @@ model = PeftModel.from_pretrained(model, "lang-uk/dragoman").to("cuda")
 tokenizer = AutoTokenizer.from_pretrained(
     "mistralai/Mistral-7B-v0.1", use_fast=False, add_bos_token=False
 )
-tokenizer.pad_token_id = tokenizer.eos_token_id
 
 
 @spaces.GPU(duration=30)
@@ -88,7 +88,7 @@ def translate(input_text):
     inputs = tokenizer([input_text], return_tensors="pt").to(model.device)
 
     generation_kwargs = dict(
-        inputs, max_new_tokens=200, num_beams=10, temperature=1
+        inputs, max_new_tokens=200, num_beams=10, temperature=1, pad_token_id=tokenizer.eos_token_id
     )  # streamer=streamer,
 
     # streaming support