Update CXR_LLAVA_HF.py

CXR_LLAVA_HF.py

@@ -584,6 +584,7 @@ class CXRLLAVAModel(PreTrainedModel):
         ]
         response = self.generate_cxr_repsonse(chat=chat, image=image, temperature=temperature, top_p=top_p)
         return response
+        
     def ask_question(self, question, image, temperature=0.2, top_p=0.8):
         chat = [
             {"role": "system",
@@ -596,7 +597,7 @@ class CXRLLAVAModel(PreTrainedModel):
 
     def generate_cxr_repsonse(self, chat, image, temperature=0.2, top_p=0.8):
         with torch.no_grad():
-            streamer = TextIteratorStreamer(self.tokenizer, skip_prompt=True, skip_special_tokens=True, timeout=15)
+            streamer = TextIteratorStreamer(self.tokenizer, skip_prompt=True, skip_special_tokens=True, timeout=90)
 
             if np.array(image).max()>255:
                 raise Exception("16-bit image is not supported.")