Solarum Asteridion commited on
Commit
b705b0b
·
verified ·
1 Parent(s): 8421f39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ class OpenAILLMHandler:
39
  self.model = os.environ.get("MODEL_NAME", "gpt-4")
40
  self.generation_lock = Lock()
41
 
42
- def generate_response(self, prompt: str, max_length: int = 512) -> str:
43
  try:
44
  with self.generation_lock:
45
  response = openai.ChatCompletion.create(
 
39
  self.model = os.environ.get("MODEL_NAME", "gpt-4")
40
  self.generation_lock = Lock()
41
 
42
+ def generate_response(self, prompt: str, max_length: int = 100) -> str:
43
  try:
44
  with self.generation_lock:
45
  response = openai.ChatCompletion.create(