Update app.py

app.py

@@ -2,18 +2,9 @@
 from moviepy.editor import *
 from PIL import Image
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM,pipeline
-import requests
 import gradio as gr
 import torch
-import re
-import os
-import sys
 from huggingface_hub import snapshot_download
-import base64
-import io
-import cv2
-import argparse
-import os
 from PIL import Image
 from min_dalle import MinDalle
 import torch
@@ -22,14 +13,10 @@ import textwrap
 from mutagen.mp3 import MP3
 # to speech conversion
 from gtts import gTTS
-from IPython.display import Audio
-from IPython.display import display
 from pydub import AudioSegment
 from os import getcwd
 import glob
 import nltk
-from IPython.display import HTML
-from base64 import b64encode
 nltk.download('punkt')
 
 description = " Video Story Generator with Audio \n PS:  Generation of video by using Artifical Intellingence by dalle-mini and distilbart and gtss "
@@ -62,7 +49,8 @@ def get_output_video(text):
           models_root=models_root,
           is_reusable=False,
           is_verbose=True,
-          dtype=torch.float16 if fp16 else torch.float32
+          #dtype=torch.float16 if fp16 else torch.float32
+          dtype=torch.float32
       )
 
       image = model.generate_image(
@@ -77,7 +65,6 @@ def get_output_video(text):
       
   generated_images = []
   for senten in plot[:-1]:
-    #print(senten)
     image=generate_image(
       is_mega='store_true',
       text=senten,
@@ -212,7 +199,7 @@ def get_output_video(text):
       audio_background = mpe.AudioFileClip(audname)
       final_clip = my_clip.set_audio(audio_background)
       final_clip.write_videofile(outname,fps=fps)
-  combine_audio(movie_name, export_path, movie_final) # i create a new file
+  combine_audio(movie_name, export_path, movie_final) # create a new file
   return 'result_final.mp4'
 text ='Once, there was a girl called Laura who went to the supermarket to buy the ingredients to make a cake. Because today is her birthday and her friends come to her house and help her to prepare the cake.'
 demo = gr.Blocks()