Update app.py

app.py

@@ -1,17 +1,19 @@
 import functools as ft
-import gradio as gr
 
+import gradio as gr
 import torch
 import transformers
 from transformers import T5ForConditionalGeneration, T5Tokenizer
 
+
 tokenizer: T5Tokenizer = T5Tokenizer.from_pretrained(
-    "models/roborovski/superprompt-v1", local_files_only=True
+    "roborovski/superprompt-v1"
 )
 model: T5ForConditionalGeneration = T5ForConditionalGeneration.from_pretrained(
-    "models/roborovski/superprompt-v1", local_files_only=True
+    "roborovski/superprompt-v1"
 )
 
+
 @ft.lru_cache(maxsize=1024)
 def super_prompt(text: str, seed: int, max_new_tokens: int, prompt: str) -> str:
     transformers.set_seed(seed)
@@ -38,6 +40,7 @@ def super_prompt(text: str, seed: int, max_new_tokens: int, prompt: str) -> str:
 
         return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
+
 demo = gr.Interface(
     fn=super_prompt,
     inputs=["text", "slider", "slider", "text"],