cv43 commited on
Commit
7403794
·
1 Parent(s): 1b6a856

initial commit

Browse files
Files changed (4) hide show
  1. .gitignore +1 -0
  2. README.md +11 -0
  3. app.py +4 -0
  4. requirements.txt +2 -1
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ __pycache__
README.md ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ tags:
4
+ - pytorch
5
+ - protocol emulation
6
+ library_name: transformers
7
+ pipeline_tag: protocol emulation
8
+ ---
9
+
10
+ # Model
11
+ This is a model fine-tuned using PyTorch Lightning...
app.py CHANGED
@@ -2,11 +2,15 @@ import gradio as gr
2
  import torch
3
  from model import Byt5LightningModule
4
 
 
5
  <<<<<<< HEAD
6
  model = torch.load("model.pt", map_location=torch.device("cpu"), weights_only=True)
7
  =======
8
  model = torch.load("model.pt", map_location=torch.device("cpu"))
9
  >>>>>>> 4524122 (initial commit)
 
 
 
10
  model.eval()
11
 
12
  def predict(input_list):
 
2
  import torch
3
  from model import Byt5LightningModule
4
 
5
+ <<<<<<< HEAD
6
  <<<<<<< HEAD
7
  model = torch.load("model.pt", map_location=torch.device("cpu"), weights_only=True)
8
  =======
9
  model = torch.load("model.pt", map_location=torch.device("cpu"))
10
  >>>>>>> 4524122 (initial commit)
11
+ =======
12
+ model = torch.load("model.pt", map_location=torch.device("cpu"), weights_only=True)
13
+ >>>>>>> 0efc206 (initial commit)
14
  model.eval()
15
 
16
  def predict(input_list):
requirements.txt CHANGED
@@ -2,4 +2,5 @@ lightning
2
  torch
3
  gradio
4
  transformers
5
- gradio
 
 
2
  torch
3
  gradio
4
  transformers
5
+ gradio
6
+ transformers