Spaces:
Running
Running
Update graph_utils.py
Browse files- graph_utils.py +1 -6
graph_utils.py
CHANGED
@@ -3,12 +3,7 @@ import gradio as gr
|
|
3 |
from transformers import pipeline
|
4 |
import os
|
5 |
|
6 |
-
model_id = models/DReAMy-lib/t5-base-DreamBank-Generation-Act-Char
|
7 |
-
|
8 |
-
title = "RE Generation"
|
9 |
-
description_GNER = """
|
10 |
-
A T5 model tuned to perform text generation, and predict which characters are present in the report. Note that, in the Hall and Van de Castle, the character lists never includes the dreamer. Hence, if you (willingly or not) enter a report that does not contain another character reference, the model will/should (correctly) produce an empty string. Moreover, it is likely that the produced list of CHAR could be longer than the one produced by the SA model, as not all CHAR might be associated with emotions.
|
11 |
-
"""
|
12 |
|
13 |
def get_graph_dict(graph_text):
|
14 |
edge_labels = {}
|
|
|
3 |
from transformers import pipeline
|
4 |
import os
|
5 |
|
6 |
+
model_id = "models/DReAMy-lib/t5-base-DreamBank-Generation-Act-Char"
|
|
|
|
|
|
|
|
|
|
|
7 |
|
8 |
def get_graph_dict(graph_text):
|
9 |
edge_labels = {}
|