Spaces:

GaborToth2
/

chatbot

Sleeping

GaborToth2 commited on Mar 12

Commit

c8b6571

1 Parent(s): 5cfff76

Implement Cohere API

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import cohere
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 COHERE_API_KEY = os.getenv("COHERE_API_KEY")
 client_cohere = cohere.Client(COHERE_API_KEY)
 COHERE_MODEL = "command-r-plus"
@@ -70,7 +70,7 @@ demo = gr.ChatInterface(
             step=0.05,
             label="Top-p (nucleus sampling)",
         ),
-        gr.Checkbox(label="Use Cohere API", default=False),
     ],
 )

 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
+client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct")
 COHERE_API_KEY = os.getenv("COHERE_API_KEY")
 client_cohere = cohere.Client(COHERE_API_KEY)
 COHERE_MODEL = "command-r-plus"
             step=0.05,
             label="Top-p (nucleus sampling)",
         ),
+        gr.Checkbox(label="Use Cohere API"),
     ],
 )