Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,9 @@ import gradio as gr
|
|
2 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
3 |
import torch
|
4 |
import spaces
|
5 |
-
|
|
|
|
|
6 |
|
7 |
def get_model_name(language):
|
8 |
"""Map language choice to the corresponding model."""
|
|
|
2 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
3 |
import torch
|
4 |
import spaces
|
5 |
+
import os
|
6 |
+
# Create the necessary directories
|
7 |
+
os.makedirs('.gradio/cached_examples/17', exist_ok=True)
|
8 |
|
9 |
def get_model_name(language):
|
10 |
"""Map language choice to the corresponding model."""
|