bat00 commited on
Commit
795e7e3
·
verified ·
1 Parent(s): d7e3f30

use gemini and google collab

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -39,6 +39,9 @@ final_answer = FinalAnswerTool()
39
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
40
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
41
 
 
 
 
42
  model = HfApiModel(
43
  max_tokens=2096,
44
  temperature=0.5,
@@ -46,6 +49,10 @@ temperature=0.5,
46
  model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
47
  custom_role_conversions=None,
48
  )
 
 
 
 
49
 
50
 
51
  # Import tool from Hub
 
39
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
40
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
41
 
42
+ '''
43
+ #I don't have any credits anymore lets use alternative way
44
+
45
  model = HfApiModel(
46
  max_tokens=2096,
47
  temperature=0.5,
 
49
  model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
50
  custom_role_conversions=None,
51
  )
52
+ '''
53
+ from google.colab import userdata
54
+ model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=userdata.get('AIzaSyBzW2rm1bR317GGQAWFHnCxVxxpIomKOHw'))
55
+
56
 
57
 
58
  # Import tool from Hub