Update app.py

app.py

@@ -1,3 +1,6 @@
+import torch 
+import tensorflow as tf 
+import flax
 import gradio as gr
 from transformers import pipeline