Sultannn commited on
Commit
ef03e92
·
1 Parent(s): 4695cc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,8 @@
3
 
4
  import gradio as gr
5
  import tensorflow as tf
 
 
6
  from transformers import MBartTokenizer, TFMBartForConditionalGeneration
7
 
8
  def run_model(input_text,
 
3
 
4
  import gradio as gr
5
  import tensorflow as tf
6
+ import torch
7
+
8
  from transformers import MBartTokenizer, TFMBartForConditionalGeneration
9
 
10
  def run_model(input_text,