mirror of
https://github.com/xtekky/gpt4free.git
synced 2024-09-11 16:55:41 +03:00
Update README.md
Added comments to clarify how to set the Hugging Face token as the API key and also improved the comments for code readability
This commit is contained in:
parent
91103e6a19
commit
92a31d8281
17
README.md
17
README.md
@ -331,9 +331,12 @@ python -m g4f.api
|
||||
```py
|
||||
import openai
|
||||
|
||||
openai.api_key = " Leave Empty if you don't use embeddings, otherwise your Hugging Face token"
|
||||
openai.api_base = "http://localhost:1337/v1"
|
||||
# Set your Hugging Face token as the API key if you use embeddings
|
||||
# If you don't use embeddings, leave it empty
|
||||
openai.api_key = "YOUR_HUGGING_FACE_TOKEN" # Replace with your actual token
|
||||
|
||||
# Set the API base URL if needed, e.g., for a local development environment
|
||||
openai.api_base = "http://localhost:1337/v1"
|
||||
|
||||
def main():
|
||||
chat_completion = openai.ChatCompletion.create(
|
||||
@ -343,18 +346,18 @@ def main():
|
||||
)
|
||||
|
||||
if isinstance(chat_completion, dict):
|
||||
# not stream
|
||||
# Not streaming
|
||||
print(chat_completion.choices[0].message.content)
|
||||
else:
|
||||
# stream
|
||||
# Streaming
|
||||
for token in chat_completion:
|
||||
content = token["choices"][0]["delta"].get("content")
|
||||
if content != None:
|
||||
if content is not None:
|
||||
print(content, end="", flush=True)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
if __name__ == "__main":
|
||||
main()
|
||||
|
||||
```
|
||||
|
||||
## Models
|
||||
|
Loading…
Reference in New Issue
Block a user