Update app.py
Browse files
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 = "
|
| 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
|
|
|
|
| 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
|