magdap116 commited on
Commit
8e93b9c
·
verified ·
1 Parent(s): 6f54a33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,7 +108,7 @@ def load_model(model_name):
108
  # model, tokenizer = load_model()
109
 
110
  #model_id = "meta-llama/Llama-3.1-8B-Instruct" # "microsoft/phi-2"# not working out of the box"google/gemma-2-2b-it" #toobig"Qwen/Qwen1.5-7B-Chat"#working but stupid: "meta-llama/Llama-3.2-3B-Instruct"
111
- model = HfApiModel()
112
  #from smolagents import TransformersModel
113
  # model = TransformersModel(
114
  # model_id=model_id,
 
108
  # model, tokenizer = load_model()
109
 
110
  #model_id = "meta-llama/Llama-3.1-8B-Instruct" # "microsoft/phi-2"# not working out of the box"google/gemma-2-2b-it" #toobig"Qwen/Qwen1.5-7B-Chat"#working but stupid: "meta-llama/Llama-3.2-3B-Instruct"
111
+ model = LiteLLMModel(model_id="anthropic/claude-3-5-sonnet-latest", temperature=0.2, max_tokens=512)
112
  #from smolagents import TransformersModel
113
  # model = TransformersModel(
114
  # model_id=model_id,