Spaces:
Sleeping
Sleeping
Update llm.py
Browse files
llm.py
CHANGED
@@ -1,22 +1,20 @@
|
|
1 |
-
import google.generativeai as genai
|
2 |
-
from
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
response = self.model.generate_content([prompt
|
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
|