Update app.py

app.py

@@ -1,11 +1,10 @@
 import gradio as gr
 from transformers import pipeline
+import os
 import torch
 
 # Set max_split_size_mb
-torch.backends.cuda.matmul.allow_tf32 = True
-torch.backends.cudnn.allow_tf32 = True
-torch.backends.cuda.matmul.max_split_size_mb = 50  # You can adjust this value
+os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:50'
 
 raven_pipeline = pipeline(
     "text-generation",