Spaces:
Running
Running
File size: 749 Bytes
60de072 df7172c 60de072 df7172c 30fc153 78e6035 60de072 30fc153 78e6035 60de072 78e6035 60de072 df7172c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# Import necessary packages
import os
from ibm_watson_machine_learning.foundation_models import Model
import gradio as gr
# Securely access the API key and project ID from environment variables
watsonx_API = os.getenv("WATSONX_API")
project_id = os.getenv("PROJECT_ID")
generate_params = {
GenParams.MAX_NEW_TOKENS: 250
}
model = Model(
model_id = 'meta-llama/llama-2-13b-chat', # you can also specify like: ModelTypes.LLAMA_2_70B_CHAT
params = generate_params,
credentials={
"apikey": watsonx_API,
"url": "https://us-south.ml.cloud.ibm.com"
},
project_id= project_id
)
q = "How to be happy?"
generated_response = model.generate(prompt=q)
print(generated_response['results'][0]['generated_text'])
|