MsChabane commited on
Commit
e8da623
·
verified ·
1 Parent(s): 6310b8b
Files changed (1) hide show
  1. main.py +3 -0
main.py CHANGED
@@ -191,9 +191,12 @@ df.dtypes:{df.dtypes.to_dict()}
191
  error.
192
  ##Prompt: {user_need}.
193
  """
 
194
  global generator
195
  output = generator(message, max_length=1000)
 
196
  match = re.search(r'```python(.*?)```', output[0]["generated_text"], re.DOTALL)
 
197
  code =''
198
  if not match:
199
  return JSONResponse(content={"error": "No Code was Generated"},status_code=403)
 
191
  error.
192
  ##Prompt: {user_need}.
193
  """
194
+ print(" MESSAGE \n"+message)
195
  global generator
196
  output = generator(message, max_length=1000)
197
+ print(F"OUTPUT {output[0]["generated_text"]}"+)
198
  match = re.search(r'```python(.*?)```', output[0]["generated_text"], re.DOTALL)
199
+
200
  code =''
201
  if not match:
202
  return JSONResponse(content={"error": "No Code was Generated"},status_code=403)