robinroy03 commited on
Commit
ba86ee1
1 Parent(s): 17aa59f

main.py modified

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -34,6 +34,6 @@ def completion():
34
  model=model,
35
  )
36
 
37
- return chat_completion.choices[0].message.content
38
 
39
- # curl -v -X POST 'http://127.0.0.1:8000/api/generate' --header 'Content-Type: application/json' --data '{"model": "llama3-70b-8192", "prompt": "why is sky blue?"}'
 
34
  model=model,
35
  )
36
 
37
+ return chat_completion.to_dict()
38
 
39
+ # curl -v -X POST 'https://robinroy03-fury-bot.hf.space/api/generate' --header 'Content-Type: application/json' --data '{"model": "llama3-70b-8192", "prompt": "why is sky blue?"}'