mirror of https://github.com/xtekky/gpt4free.git
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