Spaces:

AirrStorm
/

T5-Small-XSUM-Summarizer

Sleeping

AirrStorm commited on Jan 21

Commit

2ee6023

1 Parent(s): d1d897c

Fixing application file

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 def summarizer(sentence, min_length, max_length):
-    model_name = "/home/airrstorm/Documents/Python/Hugging-Face/Models/T5-Small-XSUM-Summarizer"
     # Create a summarization pipeline with the local model
     summarizer = pipeline("summarization", model=model_name, tokenizer=model_name)

 from transformers import pipeline
 def summarizer(sentence, min_length, max_length):
+    model_name = "AirrStorm/T5-Small-XSUM-Summarizer"
     # Create a summarization pipeline with the local model
     summarizer = pipeline("summarization", model=model_name, tokenizer=model_name)