Update README.md

README.md

@@ -48,7 +48,7 @@ import transformers
 from PIL import Image
 import sys
 
-sys.path.append("..")
+# make sure you can properly access the otter folder
 from otter.modeling_otter import OtterForConditionalGeneration
 
 # Disable warnings
@@ -156,7 +156,7 @@ def get_response(input_data, prompt: str, model=None, image_processor=None, tens
 
 
 # ------------------- Main Function -------------------
-load_bit = "fp16"
+load_bit = "fp32"
 if load_bit == "fp16":
     precision = {"torch_dtype": torch.float16}
 elif load_bit == "bf16":
@@ -174,19 +174,18 @@ image_processor = transformers.CLIPImageProcessor()
 model.eval()
 
 while True:
-    video_url = "/path/to/your_video.mp4"  # Replace with the path to your video file, could be any common format.
+    video_url = input("Enter video path: ")  # Replace with the path to your video file, could be any common format.
 
     frames_list = get_image(video_url)
 
-    prompts_input = input("Enter prompts (comma-separated): ")
-    prompts = [prompt.strip() for prompt in prompts_input.split(",")]
+    while True:
+        prompts_input = input("Enter prompts: ")
 
-    for prompt in prompts:
-        print(f"\nPrompt: {prompt}")
-        response = get_response(frames_list, prompt, model, image_processor, tensor_dtype)
-        print(f"Response: {response}")
+        if prompts_input.lower() == "quit":
+            break
 
-    if prompts_input.lower() == "quit":
-        break
+        print(f"\nPrompt: {prompts_input}")
+        response = get_response(frames_list, prompts_input, model, image_processor, tensor_dtype)
+        print(f"Response: {response}")
 
 ```