wakeupmh commited on
Commit
269dd87
·
1 Parent(s): 0499f49
Files changed (1) hide show
  1. services/model_handler.py +2 -2
services/model_handler.py CHANGED
@@ -10,7 +10,7 @@ import time
10
  import datetime
11
  import os
12
 
13
- MODEL_PATH = "google/flan-t5-base"
14
 
15
  # Simple Response class to wrap the model output
16
  class Response:
@@ -56,7 +56,7 @@ class Response:
56
  return self.content if self.content else ""
57
 
58
  def __repr__(self):
59
- return f"Response(content='{self.content[:50]}{'...' if len(self.content) > 50 else ''}')"
60
 
61
  # Personnalized class for local models
62
  class LocalHuggingFaceModel(Model):
 
10
  import datetime
11
  import os
12
 
13
+ MODEL_PATH = "google/flan-t5-large"
14
 
15
  # Simple Response class to wrap the model output
16
  class Response:
 
56
  return self.content if self.content else ""
57
 
58
  def __repr__(self):
59
+ return f"Response(content='{self.content}')"
60
 
61
  # Personnalized class for local models
62
  class LocalHuggingFaceModel(Model):