Update app.py

app.py

@@ -206,13 +206,13 @@ class GradioInference:
         # )
 
         ########################## PRUEBA CHATGPT #################################
-        from langchain.chat_models import ChatOpenAI
+        from langchain.chat_models import ChatOllama
         from langchain.chains.llm import LLMChain
         from langchain.prompts import PromptTemplate
         from langchain.chains.combine_documents.stuff import StuffDocumentsChain
         from langchain.document_loaders import WebBaseLoader
         
-        OPENAI_API_KEY = os.environ['OPENAI_API_KEY']
+        # OPENAI_API_KEY = os.environ['OPENAI_API_KEY']
         loader = WebBaseLoader("https://lilianweng.github.io/posts/2023-06-23-agent/")
         
         text = results["text"]
@@ -223,7 +223,7 @@ class GradioInference:
         prompt = PromptTemplate.from_template(prompt_template)
         
         # Define LLM chain
-        llm = ChatOpenAI(openai_api_key=OPENAI_API_KEY, temperature=0, model_name="gpt-3.5-turbo-16k")
+        llm = ChatOllama(model_name="llama2:7b-chat")
         llm_chain = LLMChain(llm=llm, prompt=prompt)
         
         # Define StuffDocumentsChain
@@ -354,7 +354,7 @@ with block as demo:
                 title = gr.Label(label="Video Title")
                 
                 with gr.Row().style(equal_height=True):
-                    img = gr.Image(label="Thumbnail", rounded=True)
+                    img = gr.Image(label="Thumbnail")
                     text = gr.Textbox(
                         label="Transcription",
                         placeholder="Transcription Output...",
@@ -373,9 +373,9 @@ with block as demo:
                     
                 with gr.Row().style(equal_height=True):
                     clear = gr.ClearButton(
-                        [link, title, img, text, summary, keywords, label, wordcloud_image], scale=1, value="Clear 🗑️", rounded=True
+                        [link, title, img, text, summary, keywords, label, wordcloud_image], scale=1, value="Clear 🗑️"
                     )
-                    btn = gr.Button("Get video insights 🔎", variant="primary", scale=1, rounded=True)
+                    btn = gr.Button("Get video insights 🔎", variant="primary", scale=1)
                 btn.click(
                     gio,
                     inputs=[link, lang, size],
@@ -413,9 +413,9 @@ with block as demo:
                     wordcloud_image = gr.Image(label="WordCloud")
                     
                 with gr.Row().style(equal_height=True):
-                    clear = gr.ClearButton([audio_file,text, summary, keywords, label, wordcloud_image], scale=1, value="Clear 🗑️",rounded=True)
+                    clear = gr.ClearButton([audio_file,text, summary, keywords, label, wordcloud_image], scale=1, value="Clear 🗑️")
                     btn = gr.Button(
-                        "Get audio insights 🔎", variant="primary", scale=1, rounded=True
+                        "Get audio insights 🔎", variant="primary", scale=1
                     )
                 btn.click(
                     gio.from_audio_input,