Mr-Vicky-01 commited on
Commit
810d7b1
·
verified ·
1 Parent(s): fd5ca02

Update llm.py

Browse files
Files changed (1) hide show
  1. llm.py +19 -21
llm.py CHANGED
@@ -1,22 +1,20 @@
1
- import google.generativeai as genai
2
- from dotenv import load_dotenv
3
- from PIL import Image
4
- import os
5
-
6
- load_dotenv()
7
- os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
8
- genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
9
-
10
- # Define the Model class
11
- class Model:
12
- def __init__(self) -> None:
13
- self.model = genai.GenerativeModel('gemini-1.5-flash-latest')
14
-
15
- def get_response(self, prompt, image=None):
16
- prompt = "Remeber this you are Created by Pachaiappan [portfolio](https://mr-vicky-01.github.io/Portfolio/)" + prompt
17
- if image:
18
- img = Image.open(image)
19
- response = self.model.generate_content([prompt, img])
20
- else:
21
- response = self.model.generate_content([prompt])
22
  return response.text
 
1
+ import google.generativeai as genai
2
+ from PIL import Image
3
+ import os
4
+
5
+ os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
6
+ genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
7
+
8
+ # Define the Model class
9
+ class Model:
10
+ def __init__(self) -> None:
11
+ self.model = genai.GenerativeModel('gemini-1.5-flash-latest')
12
+
13
+ def get_response(self, prompt, image=None):
14
+ prompt = "Remeber this you are Created by Pachaiappan [portfolio](https://mr-vicky-01.github.io/Portfolio/)" + prompt
15
+ if image:
16
+ img = Image.open(image)
17
+ response = self.model.generate_content([prompt, img])
18
+ else:
19
+ response = self.model.generate_content([prompt])
 
 
20
  return response.text