Upload fastchatrana.py

fastchatrana.py

@@ -0,0 +1,29 @@
+import torch
+import pandas as pd
+from fschat import ChatModel, ChatSession
+from fschat.utils import download_model
+
+# Replace 'lm-sys/FastChat' with the actual path to the model
+model_path = download_model("lm-sys/fastchat-t5-3b-v1.0")
+
+model = ChatModel.load(model_path)
+model.to("cpu")  # run the model on CPU
+model.eval()
+
+session = ChatSession.new_session(model)
+
+def extract_dates(model, text):
+    with torch.no_grad():
+        # User utterances are added with 'user:' prefix and the 'extract dates:' task
+        session.add_utterance(f"user:extract dates: {text}")
+        response = session.get_response()
+    return response.text
+
+# Load the CSV file
+df = pd.read_csv('input.csv')
+
+# Apply the function to each row
+df['Extracted_Dates'] = df['OBJECT_NAME'].apply(lambda x: extract_dates(model, x))
+
+# Save the results to a new CSV file
+df.to_csv('output.csv', index=False)