bugfix for main.py

images/00000-203229662.txt

@@ -1 +0,0 @@
- The image depicts a striking scene from the movie Black Swan. The ballerina, portrayed by Natalie Portman, is standing in a powerful, dramatic pose. Her body is elongated, with her arms outstretched to the sides, creating a sense of balance and tension. Her head is tilted back, gazing upwards, which adds to the intensity of the composition. The dancer's skin is pale, and her eyes are wide open, likely a deep blue or green, though the exact color is difficult to discern in this still. Her hair is dark, possibly black or dark brown, and appears to be styled in an elegant updo, though the exact style is not clear from this angle. The ballerina's body type is slender and athletic, reflecting her profession as a dancer. She is wearing a black leotard, which contrasts sharply with her pale skin and dark hair. The lighting in the scene is dramatic, with shadows playing across her face and body, emphasizing the intensity of the moment. This image captures the essence of the film's themes of obsession, dedication, and the psychological toll of pursuing perfection in dance.

images/00006-2234503665.txt

@@ -1 +0,0 @@
- The image depicts a striking scene from the movie Black Swan. The ballerina, portrayed by Natalie Portman, is standing in a powerful, dramatic pose. Her body is elongated, with her arms outstretched to the sides, creating a sense of balance and tension. Her head is tilted back, gazing upwards, which adds to the intensity of the composition. The dancer's skin is pale, and her eyes are wide open, likely a deep blue or green, though the exact color is difficult to discern in this still. Her hair is dark, possibly black or dark brown, and appears to be styled in an elegant updo, though the exact style is not clear from this angle. The ballerina's body type is slender and athletic, reflecting her profession as a dancer. She is wearing a black leotard, which contrasts sharply with her pale skin and dark hair. The lighting in the scene is dramatic, with shadows playing across her face and body, emphasizing the intensity of the moment. This image captures the essence of the film's themes of obsession, dedication, and the psychological toll of pursuing perfection in dance.

images/00030-4075734474.txt

@@ -1 +0,0 @@
- The image depicts a striking scene from the movie Black Swan. The ballerina, portrayed by Natalie Portman, is standing in a powerful, dramatic pose. Her body is elongated, with her arms outstretched to the sides, creating a sense of balance and tension. Her head is tilted back, gazing upwards, which adds to the intensity of the composition. The dancer's skin is pale, and her eyes are wide open, likely a deep blue or green, though the exact color is difficult to discern in this still. Her hair is dark, possibly black or dark brown, and appears to be styled in an elegant updo, though the exact style is not clear from this angle. The ballerina's body type is slender and athletic, reflecting her profession as a dancer. She is wearing a black leotard, which contrasts sharply with her pale skin and dark hair. The lighting in the scene is dramatic, with shadows playing across her face and body, emphasizing the intensity of the moment. This image captures the essence of the film's themes of obsession, dedication, and the psychological toll of pursuing perfection in dance.

main.py

@@ -66,6 +66,9 @@ else:
         device_map='auto',
     )
     model.to(dtype=torch.bfloat16)
+    
+generation_config = GenerationConfig(max_new_tokens=300, stop_strings="<|endoftext|>")
+bits_and_bytes_config = BitsAndBytesConfig()
 
 def unzip_images(zip_file):
     # Create a unique directory for extracted images inside the "images" directory