Spaces:
Sleeping
Sleeping
add tools
Browse files
services/model_handler.py
CHANGED
@@ -10,7 +10,7 @@ import time
|
|
10 |
import datetime
|
11 |
import os
|
12 |
|
13 |
-
MODEL_PATH = "google/flan-t5-
|
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
|
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):
|