Update app.py

app.py

@@ -5,21 +5,21 @@ import re
 # Initialize the InferenceClient with the Mixtral model
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-def translate_srt(file, target_language):
-    # Read the content of the SRT file
-    srt_content = file.read().decode("utf-8")
+def translate_srt(file_info, target_language):
+    # file_info is a dictionary containing information about the file
+    # including its content
+    srt_content = file_info["content"].decode("utf-8")
     lines = srt_content.split('\n')
     
     translated_lines = []
     for line in lines:
-        # Check if the line is a timestamp or a subtitle number
         if re.match(r"^\d+$", line) or re.match(r"^\d{2}:\d{2}:\d{2},\d{3} --> \d{2}:\d{2}:\d{2},\d{3}$", line):
             translated_lines.append(line)  # Copy timestamps and numbers directly
         elif line.strip() == "":
             translated_lines.append(line)  # Preserve empty lines for formatting
         else:
             # Translate the text line
-            response = client(inputs=line, parameters={"target_language": target_language})
+            response = client(inputs={"inputs": line, "parameters": {"target_language": target_language}})
             translated_lines.append(response[0]["generated_text"])
     
     # Join the translated lines back into a single string