Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -72,7 +72,9 @@ def read_root(request: Request, input_data: InputData):
|
|
| 72 |
]
|
| 73 |
}}
|
| 74 |
'''
|
|
|
|
| 75 |
generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
|
|
|
|
| 76 |
return {"response": generated_response}
|
| 77 |
|
| 78 |
def generate(prompt, history, temperature=0.7, max_new_tokens=30000, top_p=0.95, repetition_penalty=1.0):
|
|
|
|
| 72 |
]
|
| 73 |
}}
|
| 74 |
'''
|
| 75 |
+
print(f"{datetime.now()} - Input Text: {input_text}")
|
| 76 |
generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
|
| 77 |
+
print(f"{datetime.now()} - Response Text: {generated_response}")
|
| 78 |
return {"response": generated_response}
|
| 79 |
|
| 80 |
def generate(prompt, history, temperature=0.7, max_new_tokens=30000, top_p=0.95, repetition_penalty=1.0):
|