Update Gemini model

app.py

@@ -1,13 +1,9 @@
 import streamlit as st
-
-from transformers import pipeline
-
-from transformers import AutoTokenizer, AutoModelForCausalLM
-
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
-model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
-
-pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_length=200)
+import os
+import google.generativeai as genai
+GOOGLE_API_KEY=os.getenv('GEMINI_API_KEY')
+genai.configure(api_key=GOOGLE_API_KEY)
+model = genai.GenerativeModel(model_name = "gemini-pro")
 
 
 # Create a banner using Markdown
@@ -34,6 +30,6 @@ submit_button = st.button("Submit")
 if submit_button:
     st.write("You clicked the Submit button!")
     st.write("Entered text:", userProposal)
-    result = pipe(f"<s>[INST] {userProposal} [/INST]")
-    st.write(result)
+    response = model.generate_content([userProposal])
+    st.write(response)
 

requirements.txt

@@ -1,3 +1,4 @@
 streamlit
 altair==4.0
-transformers
+transformers
+google-generativeai