Fix deps

app.py

@@ -1,20 +1,4 @@
-from transformers import pipeline
-from functools import partial
-
-import numpy as np
-
-import matplotlib.pyplot as plt
-import seaborn as sns
-
-from sentence_transformers import SentenceTransformer
-import torch
-import tqdm
-
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, pipeline
-import penman
-from collections import Counter, defaultdict
-import networkx as nx
-from networkx.drawing.nx_agraph import pygraphviz_layout
+import gradio as gr
 
 from transformers import pipeline
 from functools import partial
@@ -129,7 +113,7 @@ class FramingDimensions:
 class FramingStructure:
     def __init__(self, base_model, roles=None):
         device = "cuda:0" if torch.cuda.is_available() else "cpu"
-        pipe2 = pipeline("text2text-generation", base_model, device=device, max_length=300)
+        self.translator = pipeline("text2text-generation", base_model, device=device, max_length=300)
 
     def __call__(self, sequence_to_translate):
         res = self.translator(sequence_to_translate)

packages.txt

@@ -1 +1,4 @@
-libgraphviz-dev
+python3-dev
+graphviz
+libgraphviz-dev
+pkg-config