Update app.py

app.py

@@ -6,7 +6,7 @@ import base64
 text_generator = pipeline("text-generation", model="Shruti9756/G24_Contract_Summarization_step3")
 
 # Increase the maximum token limit
-tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-2.7B")
+tokenizer = AutoTokenizer.from_pretrained("Shruti9756/G24_Contract_Summarization_step3")
 text_generator.model.config.max_position_embeddings = tokenizer.model_max_length
 
 # Function to generate text using the GPT-2 model
@@ -14,13 +14,13 @@ def generate_text(prompt):
     generated_text = text_generator(prompt, max_length=300, num_return_sequences=1, temperature=0.7)[0]['generated_text']
     return generated_text
 
-# Function to handle feedback and store it in a CSV file
-def handle_feedback(feedback_data, feedback_file):
-    # Your existing feedback handling logic remains the same
+# # Function to handle feedback and store it in a CSV file
+# def handle_feedback(feedback_data, feedback_file):
+#     # Your existing feedback handling logic remains the same
 
-# Function to create a download link for a binary file
-def get_binary_file_downloader_html(file_path, file_label):
-    # Your existing file downloader logic remains the same
+# # Function to create a download link for a binary file
+# def get_binary_file_downloader_html(file_path, file_label):
+#     # Your existing file downloader logic remains the same
 
 # Main Streamlit app
 def main():