WICKED4950 commited on
Commit
00d7a2e
·
verified ·
1 Parent(s): 4b69c94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
3
  print("Loading the model......")
4
  model_name = "WICKED4950/Irisonego5"
5
  strategy = tf.distribute.MirroredStrategy()
 
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
+ import tensorflow as tf
4
  print("Loading the model......")
5
  model_name = "WICKED4950/Irisonego5"
6
  strategy = tf.distribute.MirroredStrategy()