Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 import streamlit as st
-from dotenv import load_dotenv  # Import load_dotenv to load environment variables
+from dotenv import load_dotenv
 from langchain import HuggingFaceHub
 
 # Load environment variables from the .env file
@@ -15,6 +15,7 @@ def load_sql_query(question):
         # Initialize the Hugging Face model using LangChain's HuggingFaceHub class
         llm = HuggingFaceHub(
             repo_id="Salesforce/grappa_large_jnt",  # Hugging Face model repo for text-to-SQL
+            task="text2text-generation",  # Set the task to 'text2text-generation'
             huggingfacehub_api_token=HUGGINGFACE_API_TOKEN,  # Pass your API token
             model_kwargs={"temperature": 0.3}  # Optional: Adjust response randomness
         )
@@ -48,3 +49,4 @@ if submit and user_input:
     st.write(response)
 elif submit:
     st.warning("Please enter a question.")  # Warning for empty input
+