Update app.py

app.py

@@ -3,7 +3,7 @@ from pathlib import Path
 import streamlit as st
 from transformers import pipeline
 from dotenv import load_dotenv
-from langchain import PromptTemplate, HuggingFaceHub, LLMChain
+from langchain import PromptTemplate, LLMChain, OpenAI
 import requests
 
 if Path(".env").is_file():
@@ -17,7 +17,6 @@ def img2Text(url):
     st.subheader("Caption :")
     st.subheader(text)
     return text
-img2Text("photo.jpg")
 
 #llm
 def generate_story(scenario):
@@ -29,14 +28,12 @@ def generate_story(scenario):
     """
 
     prompt = PromptTemplate(template=template,input_variables=["scenario"])
-    llm_chain = LLMChain(prompt=prompt, 
-                     llm=HuggingFaceHub(repo_id="google/flan-t5-xl", 
-                                        model_kwargs={"temperature":0, 
-                                                      "max_length":64}))
-    story =llm_chain.run(scenario)
+    story_llm = LLMChain(llm=OpenAI(
+        model_name="gpt-3.5-turbo", temperature=1), prompt=prompt, verbose=True)
+    ))
+    story = story_llm.predict(scenario=scenario)
     st.subheader("Story :")
     st.subheader(story)
-    print(story)
     return story
 
 scenario = img2Text("photo.jpg")