chmawia commited on
Commit
68501e6
·
verified ·
1 Parent(s): 72cdffd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import subprocess
6
 
7
  # Force CPU usage & prevent model download issues
8
  os.environ["HF_HOME"] = "./cache" # Store model locally
9
- MODEL_NAME = "Salesforce/codet5-small"
10
 
11
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
12
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
@@ -52,4 +52,4 @@ iface = gr.Interface(
52
 
53
  # Launch the app
54
  if __name__ == "__main__":
55
- iface.launch(share=True, debug=True, server_name="0.0.0.0", server_port=7860)
 
6
 
7
  # Force CPU usage & prevent model download issues
8
  os.environ["HF_HOME"] = "./cache" # Store model locally
9
+ MODEL_NAME = "bigcode/t5-small" # Updated model
10
 
11
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
12
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
 
52
 
53
  # Launch the app
54
  if __name__ == "__main__":
55
+ iface.launch(share=True) # Simplified launch method