Update app.py

app.py

@@ -4,13 +4,12 @@ import torch
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-def load_data(path):
+def load_data(file_obj):
     """
-    Load dataset
+    Load data from the file object of the gr.File() inputs
     """
-    print(path)
-    input_path = os.path.join(path)
-    with open(input_path, "r") as f:
+    path = file_obj.name
+    with open(path, "r") as f:
         data = f.read()
 
     return data
@@ -42,7 +41,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained(config.base_model_name_or_path, de
 model = PeftModel.from_pretrained(model, peft_model_id, device_map='auto')
 
 def summarize(path):
-    transcript = load_data(path)
+    transcript = load_data(file_obj)
     texts = preprocessing(transcript)
     inputs = tokenizer(texts, return_tensors="pt", padding=True, )