Spaces:
Sleeping
Sleeping
change model
Browse files
app.py
CHANGED
|
@@ -52,10 +52,9 @@ class DummyModel:
|
|
| 52 |
self.last_input_token_count = 0
|
| 53 |
self.last_output_token_count = 0
|
| 54 |
|
| 55 |
-
def __call__(self, prompt: str):
|
| 56 |
-
|
| 57 |
-
self.
|
| 58 |
-
self.last_output_token_count = 12 # nombre fictif
|
| 59 |
|
| 60 |
return {
|
| 61 |
"text": """Thought: I now know the final answer.
|
|
|
|
| 52 |
self.last_input_token_count = 0
|
| 53 |
self.last_output_token_count = 0
|
| 54 |
|
| 55 |
+
def __call__(self, prompt: str, **kwargs):
|
| 56 |
+
self.last_input_token_count = len(prompt.split())
|
| 57 |
+
self.last_output_token_count = 10 # juste une valeur arbitraire
|
|
|
|
| 58 |
|
| 59 |
return {
|
| 60 |
"text": """Thought: I now know the final answer.
|