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'])