selvaneyas commited on
Commit
dbc7f15
·
1 Parent(s): a5efb15
Files changed (2) hide show
  1. app.py +10 -1
  2. test.py +9 -0
app.py CHANGED
@@ -31,7 +31,7 @@ if GOOGLE_API_KEY is None:
31
 
32
 
33
  # Load the Gemini Nano model
34
- model = genai.GenerativeModel("gemini-1.5-flash")
35
  #"gemini-1.5-flash" # Use "gemini-pro" or "gemini-nano" if available ""gemini-pro-vision""
36
 
37
  @app.route("/")
@@ -58,6 +58,15 @@ def services():
58
  def library():
59
  return render_template("library.html")
60
 
 
 
 
 
 
 
 
 
 
61
  #-------
62
 
63
  #-----
 
31
 
32
 
33
  # Load the Gemini Nano model
34
+ model = genai.GenerativeModel("models/gemini-1.5-flash")
35
  #"gemini-1.5-flash" # Use "gemini-pro" or "gemini-nano" if available ""gemini-pro-vision""
36
 
37
  @app.route("/")
 
58
  def library():
59
  return render_template("library.html")
60
 
61
+ @app.route("/models")
62
+ def list_models():
63
+ try:
64
+ models = genai.list_models()
65
+ return jsonify([m.name for m in models])
66
+ except Exception as e:
67
+ return jsonify({"error": str(e)}), 500
68
+
69
+
70
  #-------
71
 
72
  #-----
test.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import google.generativeai as genai
2
+ import os
3
+
4
+ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
5
+
6
+ print("Listing models…")
7
+ models = genai.list_models()
8
+ for m in models:
9
+ print(m.name)