medicare-fastapi

Sleeping

Mohammed-Altaf commited on Nov 23, 2023

Commit

f41aa83

•

1 Parent(s): bb10374

added accelerate to the requirements

Files changed (1) hide show

load_meta_data.py CHANGED Viewed

@@ -26,7 +26,7 @@ class ChatBot:
                                  # add extra parameters if models runs successfully
                                 )
         response = self.tokenizer.decode(outputs[0],skip_special_tokens=True)
-        response = self.get_clean_response(response)
         return response
     def get_clean_response(self,response):

                                  # add extra parameters if models runs successfully
                                 )
         response = self.tokenizer.decode(outputs[0],skip_special_tokens=True)
+        # response = self.get_clean_response(response)
         return response
     def get_clean_response(self,response):