Spaces:
Runtime error
Runtime error
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):
|