career_advisor / app.py
willn9's picture
Update app.py
78e6035 verified
raw
history blame
749 Bytes
# 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'])